Pep8 correction