Add support for input and output transform classes