Added probability specification