Add an export option to export the trace graph