fix main.py to work