finished part 2 of the homework