Fixed flake8 warnings