diff --git a/python/raw code/rooted_connected_k_dominating_set.py b/python/raw code/rooted_connected_k_dominating_set.py
index 82f5a06dc8e531e355ef7100c51541d3cc9b339f..c2380207a1dc91981e224c0963af31281e985891 100755
--- a/python/raw code/rooted_connected_k_dominating_set.py	
+++ b/python/raw code/rooted_connected_k_dominating_set.py	
@@ -13,6 +13,7 @@ import matplotlib.pyplot as plt
 import gurobipy as gp
 import sys
 import datetime
+import math
 
 
 def add_constraints(G, m, nodes, root):
@@ -91,5 +92,6 @@ if __name__ == '__main__':
     duration_sec = duration.total_seconds()
     print(f"iterations: {iterations}, duration(s): {duration_sec}")
     color_map = ['red' if i in ds else 'green' for i in G.nodes]
-    nx.draw(G, node_color = color_map)
+    # nx.draw(G, node_color = color_map, with_labels = True)
+    nx.draw_kamada_kawai(G, node_color = color_map)
     plt.show()
\ No newline at end of file