Flake8 suggestions