Update models.py

Use settings.AUTH_USER_MODEL rather than django.contrib.auth.models.User