Can you also update the docstrings in this function so it reflects the new format and use case?