Initial code for project_2