Update api/models.py