Add PCA, model training, cross validation and prediction on the test set