Update models.py

Use auth.get_user_model() to get the current User model