Update main.py

JK LOL