updates to trace-gen