add to edit task. Working version