Turn models.py into a module