change the main.py