Change the way to extend the User model