Now getting transformed data for train and test