fix flake8 errors
fix flake8 errors