Added support for k-fold cross validation