Validation set added with AUC measurement