update for program assignment 2