Fixed syntax error in main.py