Assignment 5 - part 2