Homework 2 finished