add update_prob lambda