Added mapping function from probability to original data