Update models.py

upgraded to custom django model