Fix pep8 complaints