add structure to get probability