Adding in the the code and updating the readme