Fixed Flake8 errors