Fix some serialization stuff

1 files changed, 12 insertions, 0 deletions

diff --git a/src/main/java/design/model/ScrambleLeague.java b/src/main/java/design/model/ScrambleLeague.java
index 92e7099..d372264 100644
--- a/src/main/java/design/model/ScrambleLeague.java
+++ b/src/main/java/design/model/ScrambleLeague.java
@@ -1,11 +1,14 @@
 package design.model;
 
 import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonTypeName;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.Date;
 import java.util.List;
 
+@JsonTypeName("scramble")
 public class ScrambleLeague extends League {
     private final List<Team> participants;
 
@@ -31,4 +34,13 @@ public class ScrambleLeague extends League {
     public Team[] getParticipants() {
         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;
+    }
 }