diff --git a/a.out b/a.out
index 9c51601..fd7bfd4 100755
Binary files a/a.out and b/a.out differ
diff --git a/graphs.cmi b/graphs.cmi
index 6828a5b..621bd70 100644
Binary files a/graphs.cmi and b/graphs.cmi differ
diff --git a/graphs.cmo b/graphs.cmo
index b67e6f3..7067315 100644
Binary files a/graphs.cmo and b/graphs.cmo differ
diff --git a/graphs.ml b/graphs.ml
index 8d75aea..290f40e 100644
--- a/graphs.ml
+++ b/graphs.ml
@@ -1252,7 +1252,7 @@ let another_type_of_dijkstra (gr : type2graph) r dx dy dt gwd ght =
     end;
   in
 
-  (* Actual BFS *)
+  (* Actual Dijkstra *)
 
   let pq = pq_create () in
 
@@ -1422,6 +1422,13 @@ let main r =
   Stdlib.print_endline "Enter the height of the graph (use 6 for weighted): ";
   let ht = Scanf.bscanf Scanf.Scanning.stdin "%d\n" identity in
 
+  Stdlib.print_endline "Enter the density of the graph (0 ~ 100): ";
+  let density = ref (Scanf.bscanf Scanf.Scanning.stdin "%d\n" identity) in
+  if(!density < 0 || !density > 100) then begin
+    Stdlib.print_endline "Invalid input, will use 50 instead";
+    density := 50;
+  end;
+
   if ht <= 0 then failwith "Error : invalid input";
 
   Stdlib.print_endline "Is the grap weighted ? (0/1)";
@@ -1441,7 +1448,7 @@ let main r =
   let offset_x = (gwd - r)/(wd-1) - 2*r in
   let offset_y = (ght - r)/(ht-1) - 2*r in
 
-  let type2 = generate_type2_graph wd ht r 1 60 in
+  let type2 = generate_type2_graph wd ht !density 1 60 in
 
   let dt = 0.25 in