Assignment 5 part 2