Update main.py

Forgot to get rid of some temporary code.