updating main.py