diff --git a/src/main/java/viewer/Display.java b/src/main/java/viewer/Display.java
index a77b7b1dc86c61f22f846698d49544296f0f1b44..1769219ead9e707913991c6195a1a5d518459f1d 100644
--- a/src/main/java/viewer/Display.java
+++ b/src/main/java/viewer/Display.java
@@ -1,13 +1,12 @@
 package viewer;
 
-import image.BlankImageFactory;
-import image.Image;
-import image.ImageFactory;
+import image.*;
 import javafx.fxml.FXML;
 import javafx.fxml.Initializable;
 import javafx.scene.canvas.Canvas;
 import javafx.scene.canvas.GraphicsContext;
 import javafx.scene.image.PixelWriter;
+import javafx.scene.paint.Color;
 
 import java.net.URL;
 import java.util.ResourceBundle;