update user model to use user trait