1 files changed, 10 insertions, 2 deletions
diff --git a/src/main/java/design/controller/userinput/menus/MainMenu.java b/src/main/java/design/controller/userinput/menus/MainMenu.java
index 7522ceb..4a221f0 100644
--- a/src/main/java/design/controller/userinput/menus/MainMenu.java
+++ b/src/main/java/design/controller/userinput/menus/MainMenu.java
@@ -2,6 +2,7 @@ package design.controller.userinput.menus;
 
 import design.controller.userinput.Menu;
 import design.controller.userinput.MenuOption;
+import design.controller.userinput.UndoActions;
 
 import java.util.List;
 
@@ -15,11 +16,18 @@ public class MainMenu extends Menu {
     public List<MenuOption> getMenuOptions() {
         return List.of(
                 new MenuOption("quit", () -> System.exit(0)),
+                new MenuOption("undo", () -> {
+                    UndoActions.undoWithSave();
+                    this.present();
+                }),
+                new MenuOption("redo", () -> {
+                    UndoActions.redoWithSave();
+                    this.present();
+                }),
                 new MenuOption("user settings...", () -> new UserSettings().present()),
                 new MenuOption("manage courses...", () -> new ManageCourses().present()),
                 new MenuOption("manage clubs...", () -> new ManageClubs().present()),
                 new MenuOption("statistics...", () -> new StatisticsMenu().present()),
-                new MenuOption("log round...", () -> new HolePlayMenu().present())
-        );
+                new MenuOption("log round...", () -> new HolePlayMenu().present()));
     }
 }