datasplit function

Split data for k-fold cross-validation