Added level completed menu state and removed debug prints

menu.lua

@@ -43,12 +43,17 @@ MenuStateIndex = {
     start = 'start',
     pause = 'pause',
     levels = 'levels',
-    hidden = 'hidden'
+    hidden = 'hidden',
+    completed = 'completed'
 }
 
+local buttonTopPos = Point:new( 240, 120 )
+local buttonBotPos = Point:new( 240, 240 )
+local buttonSize = Point:new( 320, 80 )
+
 local startButton = Button:new(
-    Point:new( 240, 120 ),
-    Point:new( 320, 80 ),
+    buttonTopPos,
+    buttonSize,
     'Start',
     nil,
     nil,
@@ -60,8 +65,8 @@ local startButton = Button:new(
 )
 
 local exitToMenuButton = Button:new(
-    Point:new( 240, 200 ),
-    Point:new( 320, 80 ),
+    buttonBotPos,
+    buttonSize,
     'Return to menu',
     nil,
     nil,
@@ -71,8 +76,8 @@ local exitToMenuButton = Button:new(
 )
 
 local exitGameButton = Button:new(
-    Point:new( 240, 240 ),
-    Point:new( 320, 80 ),
+    buttonBotPos,
+    buttonSize,
     'Exit',
     nil,
     nil,
@@ -81,6 +86,30 @@ local exitGameButton = Button:new(
     end
 )
 
+local nextLevelButton = Button:new(
+    buttonTopPos,
+    buttonSize,
+    'Next level',
+    nil,
+    nil,
+    function()
+        GameGrid = LevelHandler:next()
+        Menu.updateCellSize()
+        Menu.current_state = MenuStateIndex.hidden
+    end
+)
+
+local backToGameButton = Button:new(
+    buttonTopPos,
+    buttonSize,
+    'Continue',
+    nil,
+    nil,
+    function()
+        Menu.current_state = MenuStateIndex.hidden
+    end
+)
+
 -- Menu handler
 ---@class Menu
 ---@field states MenuState[]
@@ -93,11 +122,15 @@ Menu = {
         }),
         -- Pause menu
         [MenuStateIndex.pause] = MenuState:new({
-            exitToMenuButton
+            backToGameButton, exitToMenuButton
         }),
         -- Level selector
         [MenuStateIndex.levels] = MenuState:new({
         }),
+        -- Level completed menu
+        [MenuStateIndex.completed] = MenuState:new({
+            nextLevelButton, exitToMenuButton
+        }),
         -- Empty state 
         [MenuStateIndex.hidden] = MenuState:new()
     },