Add updateNode function