Update models.py

I made changes as suggested in models.py.