Assignment 2 part 2