small addition on documentation and pep8 fix