Need a models file, otherwise tests won't run
Change the imports to point to the new module
Remove the print statements, update the imports and add a test