project 2 updated code