DOC: imrpoved doc-string
Automatic style corrections courtesy of autopep8