cleaned up training and testing...added removal of highly correlated columns