Create problem4.rb