Update main.py

updated