Update readme and code