Modified before_save in user model