Modify addTask function