Assignment 4 part 2