Refactor code, add event when node is selected or removed