Upgrade code to use a custom user AUTH_USER_MODEL