Programming assignment 2 update