Added ROC functionality