ENH: pruning to respect flake8
Merge pull request #438 from sjsrey/flake

ENH: pruning to respect flake8