adding in probability cost