diff --git a/infinite_maze/Line.py b/infinite_maze/Line.py
index 68611a7..a996bde 100644
--- a/infinite_maze/Line.py
+++ b/infinite_maze/Line.py
@@ -95,9 +95,9 @@ def generateMaze(game, width, height):
                 lines.append(Line((xPos, yPos), (xPos, yPos + 22), sideA, sideB))
                 sideA = sideB
                 sideB += 19
-
+        
+        # Create 'maze' structure (will be complete when all 'cells' are connected to each other)
         sets = []
-        length = len(lines)
         while (len(sets) != 1):
             length = len(lines)
             lineNum = randint(0, length - 1)