Adding cross validation code..