Fix pep8 compliance