diff --git a/mnistandroid/app/src/main/java/mariannelinhares/mnistandroid/MainActivity.java b/mnistandroid/app/src/main/java/mariannelinhares/mnistandroid/MainActivity.java
index 67c02e8..81a26ab 100644
--- a/mnistandroid/app/src/main/java/mariannelinhares/mnistandroid/MainActivity.java
+++ b/mnistandroid/app/src/main/java/mariannelinhares/mnistandroid/MainActivity.java
@@ -46,6 +46,7 @@
 //encapsulates a classified image
 //public interface to the classification class, exposing a name and the recognize function
 import mariannelinhares.mnistandroid.models.Classifier;
+import mariannelinhares.mnistandroid.models.Classification;
 //contains logic for reading labels, creating classifier, and classifying
 import mariannelinhares.mnistandroid.models.TensorFlowClassifier;
 //class for drawing MNIST digits by finger
@@ -254,4 +255,4 @@ private void processTouchMove(MotionEvent event) {
     private void processTouchUp() {
         drawModel.endLine();
     }
-}
\ No newline at end of file
+}