diff --git a/src/utils/plot_utils.py b/src/utils/plot_utils.py
index bda0770..dc2905c 100644
--- a/src/utils/plot_utils.py
+++ b/src/utils/plot_utils.py
@@ -29,21 +29,9 @@ def __init__(self, config, with_title=True) -> None:
         self.config = config
         self.with_title = with_title
         self.nx_layout = None
-        # self.init_nx_graph(config)
+        self.num_users = config["num_users"]
          
 
-
-    # def init_nx_graph(self, config: ConfigType):
-    #     """
-    #     Initialize the networkx graph for the topology.
-
-    #     Args:
-    #         config (ConfigType): Configuration dictionary.
-    #         rank (int): Rank of the current node.
-    #     """
-    #     self.num_users = config["num_users"]
-    #     self.graph = nx.DiGraph()
-
     def create_adjacency_list(self, file_path: str) -> Dict[str, list]: # type: ignore
         # Load the CSV file
         """