Updated docstring to have an example