Flake8 style fixes