Add main function

This is the function being referenced from, in the setup.py. So, let's add it.

\approve @tara