Change models to match start.py