diff --git a/src/cnn/creation.c b/src/cnn/creation.c
index 4a59ffc..4b7c207 100644
--- a/src/cnn/creation.c
+++ b/src/cnn/creation.c
@@ -1,36 +1,36 @@
 #include <stdio.h>
 #include <stdlib.h>
-#include "creation.h"
-#include "function.h"
-#include "initialisation.h"
+#include "include/creation.h"
+#include "include/function.h"
+#include "include/initialisation.h"
 
 Network* create_network(int max_size, int dropout, int initialisation, int input_dim, int input_depth) {
     if (dropout < 0 || dropout > 100) {
         printf("Erreur, la probabilité de dropout n'est pas respecté, elle doit être comprise entre 0 et 100\n");
     }
-    Network* network = (Network*)malloc(sizeof(Network));
-    network->max_size = max_size;
-    network->dropout = dropout;
-    network->initialisation = initialisation;
-    network->size = 1;
-    network->input = (float****)malloc(sizeof(float***)*max_size);
-    network->kernel = (Kernel**)malloc(sizeof(Kernel*)*(max_size-1));
-    network->width = (int*)malloc(sizeof(int*)*max_size);
-    network->depth = (int*)malloc(sizeof(int*)*max_size);
+    Network* network = (Network*)malloc(sizeof(Network)); 
+    network->max_size = max_size; 
+    network->dropout = dropout; 
+    network->initialisation = initialisation; 
+    network->size = 1; 
+    network->input = (float****)malloc(sizeof(float***)*max_size); 
+    network->kernel = (Kernel**)malloc(sizeof(Kernel*)*(max_size-1)); 
+    network->width = (int*)malloc(sizeof(int*)*max_size); 
+    network->depth = (int*)malloc(sizeof(int*)*max_size); 
     for (int i=0; i < max_size; i++) {
         network->kernel[i] = (Kernel*)malloc(sizeof(Kernel));
     }
-    network->width[0] = input_dim;
-    network->depth[0] = input_depth;
-    network->kernel[0]->nn = NULL;
-    network->kernel[0]->cnn = NULL;
-    create_a_cube_input_layer(network, 0, input_depth, input_dim);
+    network->width[0] = input_dim; 
+    network->depth[0] = input_depth; 
+    network->kernel[0]->nn = NULL; 
+    network->kernel[0]->cnn = NULL; 
+    create_a_cube_input_layer(network, 0, input_depth, input_dim); 
     return network;
 }
 
 Network* create_network_lenet5(int dropout, int activation, int initialisation) {
-    Network* network = create_network(8, dropout, initialisation, 32, 1);
-    network->kernel[0]->activation = activation;
+    Network* network = create_network(8, dropout, initialisation, 32, 1); 
+    network->kernel[0]->activation = activation;  
     add_convolution(network, 6, 5, activation);
     add_average_pooling(network, 2, activation);
     add_convolution(network, 16, 5, activation);
@@ -86,28 +86,27 @@ void add_average_pooling_flatten(Network* network, int kernel_size, int activati
     network->size++;
 }
 
-void add_convolution(Network* network, int nb_filter, int kernel_size, int activation) {
+void add_convolution(Network* network, int depth_output, int kernel_size, int activation) {
     int n = network->size;
     if (network->max_size == n) {
-        printf("Impossible de rajouter une couche de convolution, le réseau est déjà plein\n");
+        printf("Impossible de rajouter une couche de convolution, le réseau est déjà plein \n");
         return;
     }
-    int r = network->depth[n-1];
-    int c = nb_filter;
+    int depth_input = network->depth[n-1];
     network->kernel[n]->nn = NULL;
     network->kernel[n]->activation = activation;
     network->kernel[n]->cnn = (Kernel_cnn*)malloc(sizeof(Kernel_cnn));
     Kernel_cnn* cnn = network->kernel[n]->cnn;
 
     cnn->k_size = kernel_size;
-    cnn->rows = r;
-    cnn->columns = c;
-    cnn->w = (float****)malloc(sizeof(float***)*r);
-    cnn->d_w = (float****)malloc(sizeof(float***)*r);
-    for (int i=0; i < r; i++) {
-        cnn->w[i] = (float***)malloc(sizeof(float**)*c);
-        cnn->d_w[i] = (float***)malloc(sizeof(float**)*c);
-        for (int j=0; j < c; j++) {
+    cnn->rows = depth_input;
+    cnn->columns = depth_output;
+    cnn->w = (float****)malloc(sizeof(float***)*depth_input);
+    cnn->d_w = (float****)malloc(sizeof(float***)*depth_input);
+    for (int i=0; i < depth_input; i++) {
+        cnn->w[i] = (float***)malloc(sizeof(float**)*depth_output);
+        cnn->d_w[i] = (float***)malloc(sizeof(float**)*depth_output);
+        for (int j=0; j < depth_output; j++) {
             cnn->w[i][j] = (float**)malloc(sizeof(float*)*kernel_size);
             cnn->d_w[i][j] = (float**)malloc(sizeof(float*)*kernel_size);
             for (int k=0; k < kernel_size; k++) {
@@ -116,9 +115,9 @@ void add_convolution(Network* network, int nb_filter, int kernel_size, int activ
             }
         }
     }
-    cnn->bias = (float***)malloc(sizeof(float**)*c);
-    cnn->d_bias = (float***)malloc(sizeof(float**)*c);
-    for (int i=0; i < c; i++) {
+    cnn->bias = (float***)malloc(sizeof(float**)*depth_output);
+    cnn->d_bias = (float***)malloc(sizeof(float**)*depth_output);
+    for (int i=0; i < depth_output; i++) {
         cnn->bias[i] = (float**)malloc(sizeof(float*)*kernel_size);
         cnn->d_bias[i] = (float**)malloc(sizeof(float*)*kernel_size);
         for (int j=0; j < kernel_size; j++) {
@@ -126,13 +125,13 @@ void add_convolution(Network* network, int nb_filter, int kernel_size, int activ
             cnn->d_bias[i][j] = (float*)malloc(sizeof(float)*kernel_size);
         }
     }
-    create_a_cube_input_layer(network, n, c, network->width[n-1] - 2*(kernel_size/2));
+    create_a_cube_input_layer(network, n, depth_output, network->width[n-1] - 2*(kernel_size/2));
     int n_int = network->width[n-1]*network->width[n-1]*network->depth[n-1];
     int n_out = network->width[n]*network->width[n]*network->depth[n];
-    initialisation_3d_matrix(network->initialisation, cnn->bias, c, kernel_size, kernel_size, n_int+n_out);
-    initialisation_3d_matrix(ZERO, cnn->d_bias, c, kernel_size, kernel_size, n_int+n_out);
-    initialisation_4d_matrix(network->initialisation, cnn->w, r, c, kernel_size, kernel_size, n_int+n_out);
-    initialisation_4d_matrix(ZERO, cnn->d_w, r, c, kernel_size, kernel_size, n_int+n_out);
+    initialisation_3d_matrix(network->initialisation, cnn->bias, depth_output, kernel_size, kernel_size, n_int+n_out);
+    initialisation_3d_matrix(ZERO, cnn->d_bias, depth_output, kernel_size, kernel_size, n_int+n_out);
+    initialisation_4d_matrix(network->initialisation, cnn->w, depth_input, depth_output, kernel_size, kernel_size, n_int+n_out);
+    initialisation_4d_matrix(ZERO, cnn->d_w, depth_input, depth_output, kernel_size, kernel_size, n_int+n_out);
     network->size++;
 }
 
diff --git a/src/cnn/free.c b/src/cnn/free.c
index 0a2dbba..9b95396 100644
--- a/src/cnn/free.c
+++ b/src/cnn/free.c
@@ -1,6 +1,6 @@
 #include <stdlib.h>
 #include <stdio.h>
-#include "free.h"
+#include "include/free.h"
 
 void free_a_cube_input_layer(Network* network, int pos, int depth, int dim) {
     for (int i=0; i < depth; i++) {
diff --git a/src/cnn/function.c b/src/cnn/function.c
index dba80df..8312cfe 100644
--- a/src/cnn/function.c
+++ b/src/cnn/function.c
@@ -1,7 +1,7 @@
 #include <stdio.h>
 #include <math.h>
 #include <float.h>
-#include "function.h"
+#include "include/function.h"
 
 float max(float a, float b) {
     return a < b ? b:a;
diff --git a/src/cnn/creation.h b/src/cnn/include/creation.h
similarity index 98%
rename from src/cnn/creation.h
rename to src/cnn/include/creation.h
index 5d7940a..54a516e 100644
--- a/src/cnn/creation.h
+++ b/src/cnn/include/creation.h
@@ -16,7 +16,7 @@ Network* create_network_lenet5(int dropout, int activation, int initialisation);
 /*
 * Créé et alloue de la mémoire à une couche de type input cube
 */
-void create_a_cube_input_layer(Network* network, int pos, int depth, int dim);
+void create_a_cube_input_layer(Network* network, int pos, int depth, int dim); // CHECKED
 
 /*
 * Créé et alloue de la mémoire à une couche de type ligne
diff --git a/src/cnn/free.h b/src/cnn/include/free.h
similarity index 100%
rename from src/cnn/free.h
rename to src/cnn/include/free.h
diff --git a/src/cnn/function.h b/src/cnn/include/function.h
similarity index 100%
rename from src/cnn/function.h
rename to src/cnn/include/function.h
diff --git a/src/cnn/initialisation.h b/src/cnn/include/initialisation.h
similarity index 100%
rename from src/cnn/initialisation.h
rename to src/cnn/include/initialisation.h
diff --git a/src/cnn/main.h b/src/cnn/include/main.h
similarity index 90%
rename from src/cnn/main.h
rename to src/cnn/include/main.h
index 5eae9b4..d194359 100644
--- a/src/cnn/main.h
+++ b/src/cnn/include/main.h
@@ -7,12 +7,12 @@
 /*
 * Renvoie si oui ou non (1 ou 0) le neurone va être abandonné
 */
-int will_be_drop(int dropout_prob);
+int will_be_drop(int dropout_prob); //CHECKED
 
 /*
 * Écrit une image 28*28 au centre d'un tableau 32*32 et met à 0 le reste
 */
-void write_image_in_network_32(int** image, int height, int width, float** input);
+void write_image_in_network_32(int** image, int height, int width, float** input); //CHECKED
 
 /*
 * Propage en avant le cnn
diff --git a/src/cnn/make.h b/src/cnn/include/make.h
similarity index 100%
rename from src/cnn/make.h
rename to src/cnn/include/make.h
diff --git a/src/cnn/struct.h b/src/cnn/include/struct.h
similarity index 100%
rename from src/cnn/struct.h
rename to src/cnn/include/struct.h
diff --git a/src/cnn/initialisation.c b/src/cnn/initialisation.c
index ef2d24a..10ff3ed 100644
--- a/src/cnn/initialisation.c
+++ b/src/cnn/initialisation.c
@@ -1,6 +1,6 @@
 #include <stdlib.h>
 #include <math.h>
-#include "initialisation.h"
+#include "include/initialisation.h"
 
 
 void initialisation_1d_matrix(int initialisation, float* matrix, int rows, int n) { //NOT FINISHED
diff --git a/src/cnn/main.c b/src/cnn/main.c
index 2561804..2aa8d23 100644
--- a/src/cnn/main.c
+++ b/src/cnn/main.c
@@ -7,7 +7,7 @@
 #include "creation.c"
 #include "make.c"
 
-#include "main.h"
+#include "include/main.h"
 
 // Augmente les dimensions de l'image d'entrée
 #define PADDING_INPUT 2
@@ -18,8 +18,8 @@ int will_be_drop(int dropout_prob) {
 
 void write_image_in_network_32(int** image, int height, int width, float** input) {
     for (int i=0; i < height+2*PADDING_INPUT; i++) {
-        for (int j=PADDING_INPUT; j < width+2*PADDING_INPUT; j++) {
-            if (i < PADDING_INPUT || i > height+PADDING_INPUT || j < PADDING_INPUT || j > width+PADDING_INPUT) {
+        for (int j=0; j < width+2*PADDING_INPUT; j++) {
+            if (i < PADDING_INPUT || i >= height+PADDING_INPUT || j < PADDING_INPUT || j >= width+PADDING_INPUT) {
                 input[i][j] = 0.;
             }
             else {
diff --git a/src/cnn/make.c b/src/cnn/make.c
index 15be0d1..7ac689b 100644
--- a/src/cnn/make.c
+++ b/src/cnn/make.c
@@ -1,5 +1,5 @@
 #include <stdio.h>
-#include "make.h"
+#include "include/make.h"
 
 void make_convolution(float*** input, Kernel_cnn* kernel, float*** output, int output_dim) {
     //NOT FINISHED, MISS CONDITIONS ON THE CONVOLUTION