Update models.py

Flake8