Updating example in the docs