Added ability to remove nodes