Fix flake8 complaints