States now only contain classes that probabilitically transition