Update models.py

More pythonic