Update models.py

test