[IMP] Add docstrings