Did some light flake8 linting.