Update homework4.rb