Update main.py

changes made