Add better docstring