Update challenge_2.rb