Update main.py

took the class out