Add use_ground_truth flag