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

Updated models.py