updated k-fold