Updated the readme to reflect the changes to the code