Update models.py

VF