Fixed Flake8 issues