Add the code and edit the readme