Updated the code and the readme.md