Programming Assignment 1 - part 3