Updated step4.rb