Programming Assignment 2 completion