Updated code basied on feed back

1 files changed, 8 insertions, 5 deletions

diff --git a/src/main/java/design/controller/userinput/menus/CourseSelectMenu.java b/src/main/java/design/controller/userinput/menus/CourseSelectMenu.java
index cf6f4b1..7cef7fc 100644
--- a/src/main/java/design/controller/userinput/menus/CourseSelectMenu.java
+++ b/src/main/java/design/controller/userinput/menus/CourseSelectMenu.java
@@ -12,6 +12,9 @@ import java.util.ArrayList;
 import java.util.List;
 import java.util.Scanner;
 
+import design.persistence.MasterDatabase;
+import design.persistence.PersonalDatabase;
+
 public class CourseSelectMenu extends Menu {
 
     @Override
@@ -23,13 +26,16 @@ public class CourseSelectMenu extends Menu {
     public List<MenuOption> getMenuOptions() {
         List<MenuOption> opts = new ArrayList<>();
 
-        // 0) pick course by number TODO: add error checking
+        // 0) back
+        opts.add(new MenuOption("back to main menu", a -> new MainMenu().present()));
+
+        // 1) pick course by number TODO: add error checking
         opts.add(new MenuOption("pick course by number (1..1000)", a -> {
             Scanner sc = new Scanner(System.in);
 
             System.out.print("Course number (1..1000): ");
             int n = Integer.parseInt(sc.nextLine().trim());
-            Course c = Session.MDB.getCourse(n - 1);
+            Course c = MasterDatabase.INSTANCE.getCourse(n - 1);
 
             System.out.print("Starting hole (1..18, blank=1): ");
             String s = sc.nextLine().trim();
@@ -42,9 +48,6 @@ public class CourseSelectMenu extends Menu {
             new HolePlayMenu(r).present();
         }));
 
-        // 1) back
-        opts.add(new MenuOption("back to main menu", a -> new MainMenu().present()));
-
         return opts;
     }
 }