adds trace_type to the models