1 files changed, 15 insertions, 16 deletions
diff --git a/src/main/java/design/model/statistics/LifetimeStats.java b/src/main/java/design/model/statistics/LifetimeStats.java
index 6e1cc97..aee4e66 100644
--- a/src/main/java/design/model/statistics/LifetimeStats.java
+++ b/src/main/java/design/model/statistics/LifetimeStats.java
@@ -1,32 +1,31 @@
 package design.model.statistics;
 
 import design.model.Golfer;
-import design.model.Play;
 import design.model.Round;
 
-public class LifetimeStats extends StatisticsDecorator{
-    private Golfer golfer;
-    public LifetimeStats(Statistics wrapped_statistics, Golfer golfer){
-        super(wrapped_statistics);
+public class LifetimeStats implements Statistics{
+    private final Golfer golfer;
+    public LifetimeStats(Golfer golfer){
         this.golfer = golfer;
     }
     @Override
+    public Round[] getRounds(){
+        return golfer.getRounds();
+    }
+    @Override
     public int get_score(){
-        int baseStats = super.get_score();
-        for(Round round : golfer.getRounds()){
-            baseStats += round.getTotalSwings();
+        int score = 0;
+        for(Round round : getRounds()){
+            score += round.getTotalSwings();
         }
-        return baseStats;
+        return score;
     }
-
     @Override
     public double get_distance(){
-        double baseStats = super.get_distance();
-        for(Round round : golfer.getRounds()){
-            for(Play play : round.getPlays()){
-                baseStats += play.getDistance();
-            }
+        double distance = 0;
+        for(Round round : getRounds()){
+            distance += round.getTotalDistance();
         }
-        return baseStats;
+        return distance;
     }
 }