Adding code for project 2