Update main.py

Fixed variable error