pep8 fixes
Updates to docs