Update models.py

#106