1 files changed, 12 insertions, 4 deletions
diff --git a/src/main/java/net/sowgro/npehero/Driver.java b/src/main/java/net/sowgro/npehero/Driver.java
index 1c6409a..fa36ee5 100755
--- a/src/main/java/net/sowgro/npehero/Driver.java
+++ b/src/main/java/net/sowgro/npehero/Driver.java
@@ -12,11 +12,8 @@ import javafx.scene.input.KeyEvent;
 import javafx.scene.layout.*;
 import javafx.stage.Stage;
 import javafx.util.Duration;
-import net.sowgro.npehero.main.Control;
+import net.sowgro.npehero.main.*;
 import net.sowgro.npehero.gui.MainMenu;
-import net.sowgro.npehero.main.Levels;
-import net.sowgro.npehero.main.Settings;
-import net.sowgro.npehero.main.Sound;
 
 import java.net.URL;
 
@@ -30,6 +27,8 @@ public class Driver extends Application
     static ImageView backgroundImage = new ImageView();
     static ImageView backgroundImage2 = new ImageView();
 
+    static Page currentPage = null;
+
     /*
      * starts javafx
      */
@@ -101,6 +100,15 @@ public class Driver extends Application
         return (Pane) primaryPane.getContent();
     }
 
+    public static void setMenu(Page p) {
+        if (currentPage != null) {
+            currentPage.onLeave();
+        }
+        currentPage = p;
+        setMenu(currentPage.getContent());
+        currentPage.onView();
+    }
+
     /**
      * Replaces the background image with a new one.
      * @param image The image to set.