Update hw0_part2.rb