Added the code and updated the readme