Small pep8 work