Update the code for new Transformation