Fix pep8 whitespace