updated regression_prob to UMASS standards