diff --git a/src/main/java/LightBoard.java b/src/main/java/LightBoard.java
index 656e0c8..6e33dd9 100644
--- a/src/main/java/LightBoard.java
+++ b/src/main/java/LightBoard.java
@@ -29,33 +29,25 @@ public LightBoard(int numRows, int numCols)
    */
   public boolean evaluateLight(int row, int col)
   {
-    int numOn = 0;
-for (int r = 0; r < lights.length; r++)
-{
-
-if (lights[r][col])
-{
-
-numOn++;
-
-}
-}
-
-if (lights[row][col] && numOn % 2 == 0)
-{
-
-return false;
-
-}
-if (!lights[row][col] && numOn % 3 == 0)
-{
-
-return true;
-
-}
-return lights[row][col];
-
-   
+    int sum = 0;
+    if(lights[row][col] == true){
+      for(int r = 0; r<lights.length;r++){
+        if(lights[r][col] == true)
+          sum++;
+      }
+      if(sum%2 == 0)
+        return false;
+      return true;
+    }
+    if(lights[row][col] == false){
+      for(int r = 0; r<lights.length;r++){
+        if(lights[r][col] == true)
+          sum++;
+      }
+      if(sum%3 == 0)
+        return true;
+      return false;
+    }
  
   }
   public boolean[][] getLights()