Update main.py

testing