Fixes to code example in Readme