change the structure of task 2, change the output structure