Clean up pep8