diff --git a/app/src/main/java/model/ColoredCellIterator.java b/app/src/main/java/model/ColoredCellIterator.java
index e558814ae259bf36fd3e844561bbee459a457932..8e846bc5dfdfbceb1091b672e5724ccff87570ba 100644
--- a/app/src/main/java/model/ColoredCellIterator.java
+++ b/app/src/main/java/model/ColoredCellIterator.java
@@ -1,9 +1,26 @@
 package model;
 
+import javafx.scene.paint.Color;
+
+import java.awt.*;
+import java.util.HashSet;
 import java.util.Iterator;
+import java.util.Set;
 
 public class ColoredCellIterator implements Iterator<Cell> {
-    public ColoredCellIterator(SquareCell squareCell) {
+    private Color color;
+    private Set<Cell> visitedCells;
+    private Set<Cell> pendingCells;
+    private Cell startCell;
+
+
+    public ColoredCellIterator(Cell startCell){
+        this.startCell=startCell;
+        visitedCells=new HashSet<>();
+        this.pendingCells=new HashSet<>();
+        this.pendingCells.add(startCell);
+        this.color = startCell.getColor();
+
     }
 
     @Override