make main a module.
easier to import in ipython shell.