Update the docs, some pep8 formating