Add trace visualization