add and update docstrings