Readme and code corrections