Modify setup_network function to add support for Node 3-5