added code to plot correlation and to fit a model by 3-fold cross validation