[ADD] docstrings for all functions