diff --git a/main.py b/main.py
index 3b9fcebb8408c29d0ad6b89240a01b1d8e9ffc42..2f39a123936b3cdeef8244a300b640b786298452 100644
--- a/main.py
+++ b/main.py
@@ -2,11 +2,12 @@ import datahandler
 import os
 from constants import EPOCHS
 
-print(EPOCHS)
-
-df_ham_train, df_ham_val = datahandler.getHamDataLoaders()
-#print(df_ham_train)
-#print(df_ham_val)
+def main():
+    df_ham_train, df_ham_val = datahandler.getHamDataLoaders()
+    #print(df_ham_train)
+    #print(df_ham_val)
 
+if __name__ == '__main__':
+    main()
 #FYI training samples for all but the 'melanocytic nevi' have been equalised (samples duplicated) so they're an even distribution
 # Before and after prints of the tables are uncommented to show (lines 142 and 154 in datahandler)
\ No newline at end of file