2 files changed, 4 insertions, 12 deletions
diff --git a/src/main/java/design/controller/userinput/menus/CourseSearch.java b/src/main/java/design/controller/userinput/menus/CourseSearch.java
index 0d5efa6..9d67b51 100644
--- a/src/main/java/design/controller/userinput/menus/CourseSearch.java
+++ b/src/main/java/design/controller/userinput/menus/CourseSearch.java
@@ -21,6 +21,7 @@ import java.util.Scanner;
 public class CourseSearch extends Menu {
     private final CurrentSearchQuery query = CurrentSearchQuery.INSTANCE;
     private final PersonalDatabase personalDB = PersonalDatabase.INSTANCE;
+    private final Golfer golfer = Session.getCurrentGolfer();
 
     @Override
     public String getTitle() {
@@ -74,18 +75,10 @@ public class CourseSearch extends Menu {
                 var name = String.format("%s, %s, Difficulty: %s, %s holes, %s total par",
                         c.getName(), c.getLocation(), c.getDifficultyRating(), c.getHoleCount(), c.getTotalPar());
                 menuOptions.add(new MenuOption(name, () -> {
-                    Golfer currentGolfer = Session.getCurrentGolfer();
-                    if (currentGolfer == null) {
-                        // if we aren't logged in, notify the user.
-                        System.out.println("\n\n !!! log into a golfer account to add courses to your profile. !!! \n\n");
-                        new MainMenu().present();
-                        return;
-                    }
-
                     // add the course, try to save to DB.
-                    currentGolfer.addCourse(c);
+                    golfer.addCourse(c);
                     try {
-                        personalDB.updateGolfer(currentGolfer);
+                        personalDB.updateGolfer(golfer);
                     } catch (IOException e) {
                         e.printStackTrace(); // not sure if we should format this prettier for the user if the DB fails.
                     }
diff --git a/src/main/java/design/controller/userinput/menus/HolePlayMenu.java b/src/main/java/design/controller/userinput/menus/HolePlayMenu.java
index aee6dd7..c824341 100644
--- a/src/main/java/design/controller/userinput/menus/HolePlayMenu.java
+++ b/src/main/java/design/controller/userinput/menus/HolePlayMenu.java
@@ -10,7 +10,6 @@ import design.model.Round;
 import design.model.holeplay.HolePlayContext;
 import design.persistence.PersonalDatabase;
 
-import java.time.LocalDateTime;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Scanner;
@@ -118,7 +117,7 @@ public class HolePlayMenu extends Menu {
         }
         // Starts round and sends user to HolePlayMenu
         Hole startHole = course.getHoles().get(startHoleNum - 1);
-        Round r = new Round(course, LocalDateTime.now(), startHole);
+        Round r = new Round(course, Session.getDateTime(), startHole);
         golfer.addRound(r);
         return r;
     }