Add feature to edit node