Include source and task id in reward