Update flake8.py