Switch docstring to numpy