Add little changes to User model