Flake8 formatting fixes