Added more features to classifier; changed the validation metric to auc_roc