Fix Flake8 issues