Put kfold cross validation in its own file