Assignment 2 Part 2 finished