Add edit task function