PCA update
Separate data into classes function