Readme and code update