Fix main to be a module