Fix pep8 warnings