Move models.py into separate files for scalability