Minor updates to example code in readme