Refer to the User model using AUTH_USER_MODEL setting
Segregate the imports and apply some PEP8 recommendations