Update main.py

Made everything great