Updated Main.py