Seperated test and training data from normalization. Added plotting of low dim PCA - can display colored clusteres of data in 2d