Could you add docstrings?