Minor changes to Main.py