1 files changed, 3 insertions, 16 deletions

diff --git a/src/main/java/design/controller/userinput/menus/MainMenu.java b/src/main/java/design/controller/userinput/menus/MainMenu.java
index 07caffa..4a221f0 100644
--- a/src/main/java/design/controller/userinput/menus/MainMenu.java
+++ b/src/main/java/design/controller/userinput/menus/MainMenu.java
@@ -2,7 +2,7 @@ package design.controller.userinput.menus;
 
 import design.controller.userinput.Menu;
 import design.controller.userinput.MenuOption;
-import design.model.undo.UndoManager;
+import design.controller.userinput.UndoActions;
 
 import java.util.List;
 
@@ -14,27 +14,14 @@ public class MainMenu extends Menu {
 
     @Override
     public List<MenuOption> getMenuOptions() {
-
-        UndoManager um = UndoManager.instance();
-
         return List.of(
                 new MenuOption("quit", () -> System.exit(0)),
                 new MenuOption("undo", () -> {
-                    if (!um.canUndo()) {
-                        System.out.println("Nothing to undo.");
-                    } else {
-                        System.out.println("Undo: " + um.peekUndoLabel());
-                        um.undo();
-                    }
+                    UndoActions.undoWithSave();
                     this.present();
                 }),
                 new MenuOption("redo", () -> {
-                    if (!um.canRedo()) {
-                        System.out.println("Nothing to redo.");
-                    } else {
-                        System.out.println("Redo: " + um.peekRedoLabel());
-                        um.redo();
-                    }
+                    UndoActions.redoWithSave();
                     this.present();
                 }),
                 new MenuOption("user settings...", () -> new UserSettings().present()),