Flake8 the code