Added probability specification generation