Finally finished the homework week 02