Added main.py

I did a TODO. Yay!!!! Main.py has almost no code, though. :(