Update problem_one.rb