Pep8ify + change on documentation