Add flag for pruning