improve prune