Create pruning.py

good result