6 files changed, 43 insertions, 1 deletions
diff --git a/src/main/java/design/model/course_search/CourseSorter.java b/src/main/java/design/model/course_search/CourseSorter.java
index 7549645..86c48a5 100644
--- a/src/main/java/design/model/course_search/CourseSorter.java
+++ b/src/main/java/design/model/course_search/CourseSorter.java
@@ -8,4 +8,5 @@ import java.util.List;
  */
 public interface CourseSorter {
     public void sortCourses(List<ICourse> courses);
+    public String toString();
 }
 \ No newline at end of file
diff --git a/src/main/java/design/model/course_search/SortByDifficulty.java b/src/main/java/design/model/course_search/SortByDifficulty.java
index 8c3251c..7b72b84 100644
--- a/src/main/java/design/model/course_search/SortByDifficulty.java
+++ b/src/main/java/design/model/course_search/SortByDifficulty.java
@@ -8,8 +8,16 @@ import java.util.Comparator;
  * Willem Dalton
  */
 public class SortByDifficulty implements CourseSorter {
+    
+    @Override
     public void sortCourses(List<ICourse> courses)
     {
         courses.sort(Comparator.comparing(ICourse::getDifficultyRating));
     }
+
+    @Override
+    public String toString()
+    {
+        return "Difficulty";
+    }
 }
 \ No newline at end of file
diff --git a/src/main/java/design/model/course_search/SortByHoles.java b/src/main/java/design/model/course_search/SortByHoles.java
index b1badab..76e2649 100644
--- a/src/main/java/design/model/course_search/SortByHoles.java
+++ b/src/main/java/design/model/course_search/SortByHoles.java
@@ -8,8 +8,17 @@ import java.util.Comparator;
  * Willem Dalton
  */
 public class SortByHoles implements CourseSorter {
+    
+    @Override
     public void sortCourses(List<ICourse> courses)
     {
         courses.sort(Comparator.comparing(ICourse::getHoleCount));
     }
-}
-\ No newline at end of file
+
+    @Override
+    public String toString()
+    {
+        return "Hole Count";
+    }
+
+}
diff --git a/src/main/java/design/model/course_search/SortByLocation.java b/src/main/java/design/model/course_search/SortByLocation.java
index 0d8a8d2..e5df839 100644
--- a/src/main/java/design/model/course_search/SortByLocation.java
+++ b/src/main/java/design/model/course_search/SortByLocation.java
@@ -8,8 +8,16 @@ import java.util.Comparator;
  * Willem Dalton
  */
 public class SortByLocation implements CourseSorter {
+    
+    @Override
     public void sortCourses(List<ICourse> courses)
     {
         courses.sort(Comparator.comparing(ICourse::getLocation));
     }
+
+    @Override
+    public String toString()
+    {
+        return "Location";
+    }
 }
 \ No newline at end of file
diff --git a/src/main/java/design/model/course_search/SortByName.java b/src/main/java/design/model/course_search/SortByName.java
index 2847d3e..cee3bcc 100644
--- a/src/main/java/design/model/course_search/SortByName.java
+++ b/src/main/java/design/model/course_search/SortByName.java
@@ -8,8 +8,16 @@ import java.util.Comparator;
  * Willem Dalton
  */
 public class SortByName implements CourseSorter {
+    
+    @Override
     public void sortCourses(List<ICourse> courses)
     {
         courses.sort(Comparator.comparing(ICourse::getName));
     }
+
+    @Override
+    public String toString()
+    {
+        return "Name";
+    }
 }
 \ No newline at end of file
diff --git a/src/main/java/design/model/course_search/SortByPar.java b/src/main/java/design/model/course_search/SortByPar.java
index 9a3eaf9..cc402b8 100644
--- a/src/main/java/design/model/course_search/SortByPar.java
+++ b/src/main/java/design/model/course_search/SortByPar.java
@@ -8,8 +8,16 @@ import java.util.Comparator;
  * Willem Dalton
  */
 public class SortByPar implements CourseSorter {
+    
+    @Override
     public void sortCourses(List<ICourse> courses)
     {
         courses.sort(Comparator.comparing(ICourse::getTotalPar));
     }
+
+    @Override
+    public String toString()
+    {
+        return "Total Par";
+    }
 }
 \ No newline at end of file