Add flake8 support and fix issues