Create prune.py