add probability class