Add tree replace node function