Update problem1.rb