Add ability to set a base_node