Updating code for the project