Add conditions to docstring