Fixed Main.py