diff --git a/app/src/main/java/model/ArrayGrid.java b/app/src/main/java/model/ArrayGrid.java
index b51a67ab612ee28f0bba6209b311f2ef68cbc754..29f17ec012347f53c118ba96083ef59decd8249d 100644
--- a/app/src/main/java/model/ArrayGrid.java
+++ b/app/src/main/java/model/ArrayGrid.java
@@ -1,5 +1,7 @@
 package model;
 
+import java.util.Iterator;
+
 public class ArrayGrid implements Grid{
 
     private Cell [][] cells;
@@ -50,4 +52,6 @@ public class ArrayGrid implements Grid{
             }
         }
     }
+
+
 }
diff --git a/app/src/main/java/model/GrayGrid.java b/app/src/main/java/model/GrayGrid.java
index 02cb8ee2f6dbc07f694cc48ba5ea0a095fabe8cd..42a49eb5d00a6dcdb562e4fac1fc1b1d20051394 100644
--- a/app/src/main/java/model/GrayGrid.java
+++ b/app/src/main/java/model/GrayGrid.java
@@ -40,4 +40,9 @@ public class GrayGrid implements Grid{
     public int getNumberOfColumns() {
         return numnberOfColumns;
     }
+
+    @Override
+    public void color(ColorGenerator colorGenerator) {
+
+    }
 }
diff --git a/app/src/main/java/model/SquareCell.java b/app/src/main/java/model/SquareCell.java
index 35f1b095d8eb861652a45589d32b8d9ffce1dba0..f8ddaff45d18f51d42f8e94e1c283e615787bce0 100644
--- a/app/src/main/java/model/SquareCell.java
+++ b/app/src/main/java/model/SquareCell.java
@@ -3,7 +3,6 @@ package model;
 import javafx.scene.paint.Color;
 
 import java.util.ArrayList;
-import java.util.Collections;
 import java.util.Iterator;
 import java.util.List;
 
@@ -11,6 +10,7 @@ public class SquareCell extends AbstractCell{
 
     List<Cell> neighbours;
 
+
     public SquareCell(){
         setColor(AbstractCell.DEFAULT_CELL_COLOR);
         setNeighbours(null);
@@ -60,4 +60,4 @@ public class SquareCell extends AbstractCell{
     }
 
 
-}
\ No newline at end of file
+}
diff --git a/app/src/test/java/model/ArrayGridTest.java b/app/src/test/java/model/ArrayGridTest.java
index b63b2e3a521518598749ae7d67ae8ac58ec5e812..543aeb043d965ed51f9d568e81954eb7bcb6ce66 100644
--- a/app/src/test/java/model/ArrayGridTest.java
+++ b/app/src/test/java/model/ArrayGridTest.java
@@ -11,9 +11,7 @@ import static org.assertj.core.api.Assertions.assertThatThrownBy;
 
 class ArrayGridTest {
 
-    // TODO
-    //  uncomment
-/*
+
     private  ArrayGrid arrayGridThreeFour;
     private final ArrayGrid arrayGridTwoTwo = new ArrayGrid(2,2);
 
@@ -76,8 +74,8 @@ class ArrayGridTest {
         }
     }
 
-    @Test
-    void testIterator() {
+   //@Test
+   /* void testIterator() {
         Iterator<Cell> iterator = arrayGridTwoTwo.iterator();
         assertThat(iterator.hasNext()).isTrue();
         assertThat(iterator.next()).isEqualTo(arrayGridTwoTwo.getCell(0,0));