Fix some pep8 problems