1 files changed, 8 insertions, 13 deletions
diff --git a/src/main/java/design/model/statistics/YearlyStats.java b/src/main/java/design/model/statistics/YearlyStats.java
index 856dda9..4c5d078 100644
--- a/src/main/java/design/model/statistics/YearlyStats.java
+++ b/src/main/java/design/model/statistics/YearlyStats.java
@@ -1,26 +1,21 @@
 package design.model.statistics;
 
-import design.model.Golfer;
+import java.util.Arrays;
+
+import design.model.Round;
 
 public class YearlyStats extends StatisticsDecorator{
     private int year;
-    private Golfer golfer;
 
-    public YearlyStats(Statistics wrapped_statistics, int year, Golfer golfer){
+    public YearlyStats(Statistics wrapped_statistics, int year){
         super(wrapped_statistics);
         this.year = year;
-        this.golfer = golfer;
-    }
-
-    @Override
-    public int get_score(){
-        int baseStats = super.get_score();
-        return baseStats;
     }
 
     @Override
-    public double get_distance(){
-        double baseStats = super.get_distance();
-        return baseStats;
+    public Round[] getRounds(){
+        return Arrays.stream(super.getRounds())
+                        .filter(round -> round.getDateTime().getYear() == year)
+                        .toArray(Round[]::new);
     }
 }