Update problem2.rb