Update main.py

z