Updated the readme of the code