1 files changed, 12 insertions, 3 deletions
diff --git a/src/main/java/design/model/Course.java b/src/main/java/design/model/Course.java
index 5f8c43d..f907622 100644
--- a/src/main/java/design/model/Course.java
+++ b/src/main/java/design/model/Course.java
@@ -1,8 +1,12 @@
 package design.model;
 
+import design.model.course_search.ICourse;
+
 import java.util.List;
 
-public class Course {
+public class Course implements ICourse {
+
+    private final int id;
     private final String name;
     private final int difficultyRating;
     private final String location;
@@ -10,7 +14,8 @@ public class Course {
     private final int totalPar;
     private final List<Hole> holes;
 
-    public Course(String name, int difficultyRating, String location, int holeCount, int totalPar, List<Hole> holes) {
+    public Course(int id, String name, int difficultyRating, String location, int holeCount, int totalPar, List<Hole> holes) {
+        this.id = id;
         this.name = name;
         this.difficultyRating = difficultyRating;
         this.location = location;
@@ -19,11 +24,15 @@ public class Course {
         this.holes = holes;
     }
 
+    public int getId() {
+        return id;
+    }
+
     public String getName() {
         return name;
     }
 
-    public int getDifficultyRating() {
+    public float getDifficultyRating() {
         return difficultyRating;
     }