diff --git a/src/main/java/NTNU/IDATT1002/repository/ImageRepository.java b/src/main/java/NTNU/IDATT1002/repository/ImageRepository.java
index b2f737184613662ea25f9c80495e3f448ef8028c..c22a18f7250c8e3c9b897d4b0b070b8fb1c07f45 100644
--- a/src/main/java/NTNU/IDATT1002/repository/ImageRepository.java
+++ b/src/main/java/NTNU/IDATT1002/repository/ImageRepository.java
@@ -1,9 +1,7 @@
 package NTNU.IDATT1002.repository;
 
 import NTNU.IDATT1002.models.Image;
-import java.util.Optional;
-import javax.persistence.EntityManager;
-import java.util.List;
+  import javax.persistence.EntityManager;
 
 
 /**
@@ -18,8 +16,6 @@ import java.util.List;
 
 public class ImageRepository extends GenericRepository<Image, Long> {
 
-  private EntityManager entityManager;
-
   /**
    * Constructor to inject {@link EntityManager} dependency.
    *
@@ -27,28 +23,10 @@ public class ImageRepository extends GenericRepository<Image, Long> {
    */
   public ImageRepository(EntityManager entityManager) {
     super(entityManager);
+    setClassType(Image.class);
   }
+}
 
 
-  /**
-   * Retrieves all instances of the type image.
-   *
-   * @return all entities
-   */
-  public Optional<Image> update(Image image) {
-    return Optional.empty();
-  }
-
-
-  /**
-   * Return whether the given image exists.
-   *
-   * @param image image album to check existence for
-   * @return true if the image album exist, else false
-   */
 
 
-  public boolean exists(Image image) {
-    return findById(image.getImageID()).isPresent();
-  }
-}