Adding transform function to transition kernel