Add docstring to public functions