1 files changed, 15 insertions, 0 deletions
diff --git a/src/main/java/design/model/ScrambleLeague.java b/src/main/java/design/model/ScrambleLeague.java
index 55f9893..1fb8f1d 100644
--- a/src/main/java/design/model/ScrambleLeague.java
+++ b/src/main/java/design/model/ScrambleLeague.java
@@ -1,6 +1,7 @@
 package design.model;
 
 import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonTypeName;
 
 import java.util.ArrayList;
 import java.util.Collections;
@@ -10,6 +11,7 @@ import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 
+@JsonTypeName("scramble")
 public class ScrambleLeague extends League {
     private List<Team> participants;
     private Map<Team, Integer> totalTeamScores;
@@ -43,6 +45,19 @@ public class ScrambleLeague extends League {
         return participants.toArray(Team[]::new);
     }
 
+    public Team locateTeam(Golfer golfer) {
+        for (Team participant : participants) {
+            if (List.of(participant.getMembers()).contains(golfer)) {
+                return participant;
+            }
+        }
+        return null;
+    }
+
+    @Override
+    public String getType() {
+        return "scramble";
+    }
     @Override
     public void recordPlay(Golfer player, Match match, Round round){
         if(!isPlayable()) return;