Update main.py

Fixed error on line 12.