add prob output
add prob output