1 files changed, 24 insertions, 11 deletions

diff --git a/src/main/java/design/controller/userinput/menus/HolePlayMenu.java b/src/main/java/design/controller/userinput/menus/HolePlayMenu.java
index 2a7fc5d..6651500 100644
--- a/src/main/java/design/controller/userinput/menus/HolePlayMenu.java
+++ b/src/main/java/design/controller/userinput/menus/HolePlayMenu.java
@@ -11,6 +11,7 @@ import design.model.Round;
 import design.model.holeplay.HolePlayContext;
 import design.persistence.PersonalDatabase;
 import design.model.undo.UndoManager;
+import design.controller.userinput.UndoActions;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -37,17 +38,37 @@ public class HolePlayMenu extends Menu {
     public List<MenuOption> getMenuOptions() {
         List<MenuOption> opts = new ArrayList<>();
 
+        // End round (always shown) 1
+        opts.add(new MenuOption("end round", () -> {
+            UndoManager.instance().capture(golfer, "End round on " + round.getCourse().getName());
+            ctx.endRoundNow();
+            System.out.println("Round ended.");
+            new MainMenu().present();
+        }));
+
+        // 2
+        opts.add(new MenuOption("undo", () -> {
+            UndoActions.undoWithSave();
+            this.present();
+        }));
+
+        // 3
+        opts.add(new MenuOption("redo", () -> {
+            UndoActions.redoWithSave();
+            this.present();
+        }));
+
         boolean inSetup = (ctx.getCurrentPlay() == null);
 
         if (inSetup) {
-            // 0) Start hole
+            // 4) Start hole
             opts.add(new MenuOption("start hole", () -> {
                 ctx.startHole();
                 System.out.println("Started hole " + round.getCurrentHole().getNumber() + ".");
                 this.present();
             }));
         } else {
-            // 0) Take a shot
+            // 4) Take a shot
             opts.add(new MenuOption("take a shot", () -> {
                 var selector = new SelectClub();
                 selector.present();
@@ -72,7 +93,7 @@ public class HolePlayMenu extends Menu {
                 this.present();
             }));
 
-            // 1) Hole out
+            // 5) Hole out
             opts.add(new MenuOption("hole out", () -> {
                 // Precedes to next hole
                 int prev = round.getCurrentHole().getNumber();
@@ -85,14 +106,6 @@ public class HolePlayMenu extends Menu {
             }));
         }
 
-        // End round (always shown)
-        opts.add(new MenuOption("end round", () -> {
-            UndoManager.instance().capture(golfer, "End round on " + round.getCourse().getName());
-            ctx.endRoundNow();
-            System.out.println("Round ended.");
-            new MainMenu().present();
-        }));
-
         return opts;
     }