Fix flake8 warnings