diff --git a/main.c b/main.c
index 82750fe..0330abf 100644
--- a/main.c
+++ b/main.c
@@ -1,10 +1,12 @@
 #include <stdio.h>
 #include <stdio.h>
+#include <malloc.h>
 
 #include "matrix.h"
 #include "image.h"
 
 int main() {
-    Image** images = import_images("../data/train-images.idx3-ubyte", "../data/train-labels.idx1-ubyte", 20);
+    Image** images = import_images("../data/train-images.idx3-ubyte", "../data/train-labels.idx1-ubyte", NULL, 2);
+    img_visualize(images[1]);
 
 }
\ No newline at end of file