1 files changed, 9 insertions, 10 deletions
diff --git a/src/main/java/design/model/statistics/RoundStats.java b/src/main/java/design/model/statistics/RoundStats.java
index f65a249..598ca59 100644
--- a/src/main/java/design/model/statistics/RoundStats.java
+++ b/src/main/java/design/model/statistics/RoundStats.java
@@ -1,22 +1,21 @@
 package design.model.statistics;
 
+import java.util.Arrays;
+
 import design.model.Round;
 
 public class RoundStats extends StatisticsDecorator{
-    private Round round;
+    private Round target_round;
 
-    public RoundStats(Statistics wrapped_statistics, Round round){
+    public RoundStats(Statistics wrapped_statistics, Round target_round){
         super(wrapped_statistics);
-        this.round = round;
-    }
-
-    @Override
-    public int get_score(){
-        return super.get_score();
+        this.target_round = target_round;
     }
 
     @Override
-    public double get_distance(){
-        return super.get_distance();
+    public Round[] getRounds(){
+        return Arrays.stream(super.getRounds())
+                        .filter(round -> round == target_round)
+                        .toArray(Round[]::new);
     }
 }