fix comment in main.py