Got a missing separator error and its late.
Wrap main into a function.

This is mostly to silence pylint warnings. I raised my grade by one
point simpy by doing it :)