Finish HW part 1
HW 2