Update problem_1.rb