Update setup_federated_node.py