Updated PCA code to use the one in sklearn.