fixed error in Main.py