Add assignment 3 part 2