impr(task diagram): change the highlighting of task

Related to: CAM-3017