Adding remove node feature