Get CV and Grid Search working