fix Readme and code