Added single branch pruning