Updating code example in readme.