Final touches on add_node tests