Updating main.py