Added trainTestAUC