Update HW_01.rb