1 files changed, 15 insertions, 16 deletions
diff --git a/src/main/java/design/controller/userinput/menus/StatisticsMenu.java b/src/main/java/design/controller/userinput/menus/StatisticsMenu.java
index c299d3e..16d875a 100644
--- a/src/main/java/design/controller/userinput/menus/StatisticsMenu.java
+++ b/src/main/java/design/controller/userinput/menus/StatisticsMenu.java
@@ -57,6 +57,10 @@ public class StatisticsMenu extends Menu {
         return null;
     }
 
+    private Statistics getUserHole(Statistics baseStatistics){
+        return null;
+    }
+
     @Override
     public List<MenuOption> getMenuOptions() {
         return List.of(
@@ -86,27 +90,22 @@ public class StatisticsMenu extends Menu {
                 }),
                 new MenuOption("view round statistics", () -> {
                     Statistics baseStats = new LifetimeStats(golfer);
-                    Scanner sc = new Scanner(System.in);
-
-                    System.out.println("Enter year to search: ");
-                    int year = sc.nextInt();
-                    Statistics yearStats = new YearlyStats(baseStats, year);
+                    Statistics yearStats = getUserYear(baseStats);
+                    Statistics courseStats = getUserCourse(yearStats);
+                    Statistics roundStats = getUserRound(courseStats);
                     
-                    System.out.printf("Total swings: %d\n", yearStats.get_score());
-                    System.out.printf("Total distance: %.1f\n", yearStats.get_distance());
-                    sc.close();
+                    System.out.printf("Total swings: %d\n", roundStats.get_score());
+                    System.out.printf("Total distance: %.1f\n", roundStats.get_distance());
                 }),
                 new MenuOption("view hole statistics", () -> {
                     Statistics baseStats = new LifetimeStats(golfer);
-                    Scanner sc = new Scanner(System.in);
-
-                    System.out.println("Enter year to search: ");
-                    int year = sc.nextInt();
-                    Statistics yearStats = new YearlyStats(baseStats, year);
+                    Statistics yearStats = getUserYear(baseStats);
+                    Statistics courseStats = getUserCourse(yearStats);
+                    Statistics roundStats = getUserRound(courseStats);
+                    Statistics holeStats = getUserHole(roundStats);
                     
-                    System.out.printf("Total swings: %d\n", yearStats.get_score());
-                    System.out.printf("Total distance: %.1f\n", yearStats.get_distance());
-                    sc.close();
+                    System.out.printf("Total swings: %d\n", holeStats.get_score());
+                    System.out.printf("Total distance: %.1f\n", holeStats.get_distance());
                 })
         );
     }