Update models.py

Pull request to fix #53
Merge pull request #54 from slollo/patch-1

Update models.py