modify implementation of class in the User model