Enhanced Transform stage