Trying to deal with N_samples < N_folds

Needs to be fixed more