1 files changed, 11 insertions, 9 deletions
diff --git a/src/gui/LevelSelector.java b/src/gui/LevelSelector.java
index 25d1ed2..a84444b 100644
--- a/src/gui/LevelSelector.java
+++ b/src/gui/LevelSelector.java
@@ -4,23 +4,25 @@ import javafx.collections.FXCollections;
 import javafx.collections.ObservableList;
 import javafx.geometry.Pos;
 import javafx.scene.Scene;
+import javafx.scene.control.Button;
 import javafx.scene.control.ListView;
+import javafx.scene.layout.Pane;
 import javafx.scene.layout.VBox;
 import javafx.stage.Stage;
 
-public class LevelSelector extends Scene
-{
-    private static VBox centerMenu2 = new VBox();
-    
-    public LevelSelector(Stage primaryStage)
+public class LevelSelector extends Pane
+{   
+    public LevelSelector()
     {
-        super(centerMenu2,800,600);
-        primaryStage.setTitle("NPE Hero - Level Selector");
         ListView<String> levels = new ListView<String>();
         ObservableList<String> levelList= FXCollections.observableArrayList ("Test Level 1", "Test Level 2", "Test Level 3", "Test Level 4");
         levels.setItems(levelList);
-        centerMenu2.setAlignment(Pos.CENTER);
-        centerMenu2.getChildren().add(levels);
+        //super.setAlignment(Pos.CENTER);
+        Button exit = new Button();
+        exit.setText("Exit");
+        exit.setOnAction(e -> Driver.switchMenu("MainMenu"));
+
+        super.getChildren().addAll(levels,exit);
     }
 
 }