ENH: pruning to respect flake8