nit: end docstring on separate line e.g.

```python
"""This function does this
and this 
"""