modify content of main.py