update to main.py