Fix flake8 errors
Update flake8 config