Readjsuted the docstrings.

Per feedback from @mrocklin.