Pep8 fix and updated docs