Fixed main.py

I really need to remember to not use testing code in my Github copy…