DOC: Clean docstrings and add example