updating code to specs of project