flake8 style fixes