Updated docstring and example in README