Added another code for the homework 1.