Fix pep8 formatting