Improve models.py.