Updates to pca calculations and plotting