Programing assignment 2 update