Readme and code updated