Fixed flake8 formatting