updated k-fold
updated k-fold