Add docstring

As per review feedback.