Fix Flake8 errors