diff --git a/src/mnist/include/preview.h b/src/mnist/include/preview.h
index 32f91a8..b8d9ccd 100644
--- a/src/mnist/include/preview.h
+++ b/src/mnist/include/preview.h
@@ -6,9 +6,18 @@
 #ifndef DEF_PREVIEW_H
 #define DEF_PREVIEW_H
 
+/*
+* Affiche un chiffre de taille width x height
+*/
 void print_image(unsigned int width, unsigned int height, int** image);
+
+/* 
+* Prévisualise un chiffre écrit à la main
+*/
 void preview_images(char* images_file, char* labels_file);
 
+int main(int argc, char *argv[]);
+
 #endif
 
                
\ No newline at end of file
diff --git a/src/mnist/preview.c b/src/mnist/preview.c
index 60acdb3..058ab78 100644
--- a/src/mnist/preview.c
+++ b/src/mnist/preview.c
@@ -7,8 +7,6 @@
 #include "include/preview.h"
 
 
-// Prévisualise un chiffre écrit à la main
-// de taille width x height
 void print_image(unsigned int width, unsigned int height, int** image) {
     char tab[] = {' ', '.', ':', '%', '#', '\0'};