adited lifetime stats

2 files changed, 10 insertions, 4 deletions

diff --git a/src/main/java/design/model/statistics/LifetimeStats.java b/src/main/java/design/model/statistics/LifetimeStats.java
index 9cefd7a..6e1cc97 100644
--- a/src/main/java/design/model/statistics/LifetimeStats.java
+++ b/src/main/java/design/model/statistics/LifetimeStats.java
@@ -5,7 +5,7 @@ import design.model.Play;
 import design.model.Round;
 
 public class LifetimeStats extends StatisticsDecorator{
-    Golfer golfer;
+    private Golfer golfer;
     public LifetimeStats(Statistics wrapped_statistics, Golfer golfer){
         super(wrapped_statistics);
         this.golfer = golfer;
diff --git a/src/main/java/design/model/statistics/YearlyStats.java b/src/main/java/design/model/statistics/YearlyStats.java
index 026e2c5..856dda9 100644
--- a/src/main/java/design/model/statistics/YearlyStats.java
+++ b/src/main/java/design/model/statistics/YearlyStats.java
@@ -1,20 +1,26 @@
 package design.model.statistics;
 
+import design.model.Golfer;
+
 public class YearlyStats extends StatisticsDecorator{
     private int year;
+    private Golfer golfer;
 
-    public YearlyStats(Statistics wrapped_statistics, int year){
+    public YearlyStats(Statistics wrapped_statistics, int year, Golfer golfer){
         super(wrapped_statistics);
         this.year = year;
+        this.golfer = golfer;
     }
 
     @Override
     public int get_score(){
-        return super.get_score();
+        int baseStats = super.get_score();
+        return baseStats;
     }
 
     @Override
     public double get_distance(){
-        return super.get_distance();
+        double baseStats = super.get_distance();
+        return baseStats;
     }
 }