Fix some pep8 stuff