Trace-for-imitation-models