diff --git a/ImageRecognitionNN.py b/ImageRecognitionNN.py
index 0cdfed9..5983337 100644
--- a/ImageRecognitionNN.py
+++ b/ImageRecognitionNN.py
@@ -6,12 +6,12 @@
 from progress.bar import IncrementalBar
 
 class_names = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']
-show_testing = True;
+show_testing = True
 
 class NeuralNetwork():
 
     width = 40
-    epochs = 2
+    epochs = 1
     numUsing = 60000
     bs = 250
     beta1 = 0.9
@@ -191,8 +191,7 @@ def predict(self, testImage):
         self.forwardProp(testImage)
         return np.argmax(self.outputNeurons)
 
-if __name__ == "__main__":
-
+def start():
     (train_images, train_labels), (test_images, test_labels) = mnist.load_data()
     train_images = train_images/255
 
diff --git a/__pycache__/ImageRecognitionNN.cpython-38.pyc b/__pycache__/ImageRecognitionNN.cpython-38.pyc
new file mode 100644
index 0000000..b368e8c
Binary files /dev/null and b/__pycache__/ImageRecognitionNN.cpython-38.pyc differ
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..32f604e
--- /dev/null
+++ b/main.py
@@ -0,0 +1,3 @@
+from ImageRecognitionNN import start
+
+start()
\ No newline at end of file