Completed code for Programming Assignment 2