Getting closer to being able to generate roc curves again