Add trace collector set code