edit main.py again