Update hw1_part4.rb