diff --git a/src/a_star.py b/src/a_star.py
index 60a98567bdc3412cd3f7f43c8db671997122be00..4d3734eb1eafd65ac6aab05509ab54cb976dee1e 100644
--- a/src/a_star.py
+++ b/src/a_star.py
@@ -42,6 +42,8 @@ def a_star(graph, start_node, goal_node, heuristic):
 graph, start_node, goal_node = get_graph_1()
 path = a_star(graph, start_node, goal_node, max_cost)
 print(f"Start node: {start_node.id}. Goal node: {goal_node.id}")
+print("-" * 5, end="")
+print("Solution plan", end="")
+print("-" * 5)
 for x in path:
     print(x.id)
-print("-" * 30)
diff --git a/src/graph_tests.py b/src/graph_tests.py
index 8310610b3571394fdc959996ab7c05259c314632..29c0e035b838c5dcb73c0eb37d408cdc3b3ccdca 100644
--- a/src/graph_tests.py
+++ b/src/graph_tests.py
@@ -7,12 +7,16 @@ def get_graph_1():
     node_2 = Node(2)
     node_3 = Node(3)
     node_4 = Node(4)
+    node_5 = Node(5)
+    node_6 = Node(6)
 
     graph.set_bidirectional_neighbor(node_1, node_2, 1)
     graph.set_bidirectional_neighbor(node_1, node_3, 1)
     graph.set_bidirectional_neighbor(node_2, node_4, 1)
     graph.set_bidirectional_neighbor(node_3, node_4, 1)
+    graph.set_bidirectional_neighbor(node_3, node_5, 1)
+    graph.set_bidirectional_neighbor(node_5, node_6, 1)
 
-    graph.add_nodes([node_1, node_2, node_3, node_4])
+    graph.add_nodes([node_1, node_2, node_3, node_4, node_5, node_6])
 
-    return graph, node_1, node_4
+    return graph, node_1, node_6