DOC: Add parameter to docstring