Pep8 fixed