updated for flake8