Finished HW_01.rb