diff: apply flake8 suggestions