Fix Flake8 violations