Add SourceWrapper and SinkWrapper