Added missing prune feature