Add setNodes function