Now keeping track of pruned states