Update hw1_part6.rb