update Prog Assign 2