Add tests for flake8