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

Update models.py