Update HW1-4.rb