Changes to Main.py