add trait name on model output page