Removed unnecessary code in Main.py