Update models.py
Merge pull request #5 from alexxxxey/patch-1

Update models.py