1 files changed, 10 insertions, 10 deletions
diff --git a/src/main/java/design/model/statistics/HoleStats.java b/src/main/java/design/model/statistics/HoleStats.java
index 64fa96c..2c023b4 100644
--- a/src/main/java/design/model/statistics/HoleStats.java
+++ b/src/main/java/design/model/statistics/HoleStats.java
@@ -1,22 +1,22 @@
 package design.model.statistics;
 
+import java.util.Arrays;
+
 import design.model.Hole;
+import design.model.Round;
 
 public class HoleStats extends StatisticsDecorator{
-    private Hole hole;
+    private Hole target_hole;
 
-    public HoleStats(Statistics wrapped_statistics, Hole hole){
+    public HoleStats(Statistics wrapped_statistics, Hole target_hole){
         super(wrapped_statistics);
-        this.hole = hole;
-    }
-
-    @Override
-    public int get_score(){
-        return super.get_score();
+        this.target_hole = target_hole;
     }
 
     @Override
-    public double get_distance(){
-        return super.get_distance();
+    public Round[] getRounds(){
+        return Arrays.stream(super.getRounds())
+                        .filter(round -> round.getCourse().getHoles().stream().anyMatch(hole -> hole.equals(target_hole)))
+                        .toArray(Round[]::new);
     }
 }