Updated WeightedRandomGenerator