1 files changed, 12 insertions, 0 deletions
diff --git a/src/main/java/design/model/Hole.java b/src/main/java/design/model/Hole.java
index bc9fdb9..f79cc3d 100644
--- a/src/main/java/design/model/Hole.java
+++ b/src/main/java/design/model/Hole.java
@@ -3,6 +3,7 @@ package design.model;
 public class Hole {
     private final int number;
     private final int par;
+    // TODO: Maybe add a yardage interger var
 
     public Hole(int number, int par) {
         this.number = number;
@@ -16,4 +17,15 @@ public class Hole {
     public int getPar() {
         return par;
     }
+
+    @Override
+    public boolean equals(Object obj){
+        if(!(obj instanceof Hole)){
+            return false;
+        }
+        Hole object = (Hole)obj;
+        return (object.getNumber() == this.getNumber() && object.getPar() == this.getPar());
+    }
+
+    // TODO: Maybe add a get yardage method
 }