diff --git a/mini_proj/test_nn.py b/mini_proj/test_nn.py
index 0d8ec2e..987b6b5 100644
--- a/mini_proj/test_nn.py
+++ b/mini_proj/test_nn.py
@@ -56,15 +56,17 @@ if __name__ == '__main__':
     image = cv2.imread("10.jpg")
     # Split image
     cuts = image_cut(image, 64, 64)
-    for i in len(cuts):
+    for i in range(len(cuts)):
         # Transform block
         hsv = color.rgb2hsv(cuts[i])
         hsv[:, :, 2] = exposure.equalize_hist(hsv[:, :, 2])
         block = color.hsv2rgb(hsv)
         block = np.rollaxis(block, -1)
         if is_Wally("Waldo.h5", block):
+        # if True:
             # Border block
-            cuts[i] = cv2.copyMakeBorder(cuts[i],5,5,5,5,cv2.BORDER_CONSTANT,value=RED)
+            GREEN = [0, 255, 0]
+            cuts[i] = cv2.copyMakeBorder(cuts[i][1:61,1:61],2,2,2,2,cv2.BORDER_CONSTANT,value=GREEN)
 
     # Stitch image TODO!
     # Show image