assign.folds function

Assign observations to folds in a balanced way