Update problem4.rb