Added cross-validation to train