Added simple pruning