DOC: add  to the docstring where needed