Updated step3.rb to edit in new way