flake8 style cleanup