EDIT: Revised the itk pipeline.