Programming Assignment 2 Update