update Programming Assignment 2