Add `Returns` to docstring?