Fix new pep257 errors, updated docstring