Updating Programming 2 assignment