Separating functions: Create Folds and Dataset class
Polished code