update code for project2