Added solution to part 2 of the assignment