Please update the docstring and examples with the new arguments.