Assignment 1 part 2 - done