Add docstring
Edit to pass lint