Update Main.py

Fixed a small bug.