Some cleanups of include/dai/prob.h