add code to user.rb