Move AST transformations to OpTree