Fix to prdiction bug, ROC is much better now