main.py edited