Add book covers

1 files changed, 8 insertions, 8 deletions

diff --git a/src/main/scala/reading/models/Filter.scala b/src/main/scala/reading/models/Filter.scala
index b8adcc1..00c7994 100644
--- a/src/main/scala/reading/models/Filter.scala
+++ b/src/main/scala/reading/models/Filter.scala
@@ -10,9 +10,9 @@ sealed trait FilterKind
 case object PeriodKind extends FilterKind
 case object ThemeKind extends FilterKind
 case object GenreKind extends FilterKind
-case object DifficultyKind extends FilterKind
-case object ProgramKind extends FilterKind
 case object LevelKind extends FilterKind
+case object ProgramKind extends FilterKind
+case object GradeKind extends FilterKind
 
 object Filter {
   def apply[T](in: T)(implicit filterFactory: FilterFactory[T]): Filter =
@@ -26,7 +26,7 @@ object Filter {
 
   def remove(fs: Seq[Filter], rf: Filter): Seq[Filter] =
     fs.filterNot { f =>
-      equals(f, rf) || rf.kind == LevelKind && f.kind == ProgramKind
+      equals(f, rf) || rf.kind == GradeKind && f.kind == ProgramKind
     }
 }
 
@@ -71,12 +71,12 @@ object FilterFactory {
       }
   }
 
-  implicit object LevelFilter extends FilterFactory[Level] {
-    def create(level: Level): Filter =
+  implicit object GradeFilter extends FilterFactory[Grade] {
+    def create(grade: Grade): Filter =
       new Filter {
-        def filter(book: Book): Boolean = book.programs.map(Program.level).contains(level)
-        val kind: FilterKind = LevelKind
-        val name: String = level.toString()
+        def filter(book: Book): Boolean = book.programs.map(Program.grade).contains(grade)
+        val kind: FilterKind = GradeKind
+        val name: String = grade.toString()
       }
   }
 }