Custom User model changes.

I had to make the following changes for my custom user model