Update main.py

oops.. forgot to put __main__ as a string