Update main.py

I made a change on line 129