add numpydoc style docstring