Some files needed to be added back to the project for the code to work