Allow cv.folds to be set as argument