Update models.py
Merge pull request #15 from jlin45/patch-12

Update models.py