Update problem_2.rb