You mask out user prompts / tool call responses in training