Update problem_4.rb