Removed commented code, added prediction to Main.py