Add the code and update the readme