Finished the homework assignment HW_01.rb