Fix flake8 violations