edit to code example in Readme