Updated main.py to execute without syntax errors