Update Main.py

Small fix