Change models to be a module