make_folds function

Create Folds for Cross-Fitting