Update models.py

I think the use of the built-in User class is more powerful.