diff --git a/doc/design/images/graph_construction_example.dot b/doc/design/images/graph_construction_example.dot
index 86f82fe05af2b8b77ed9b25a5f39562456eb7ec1..bedb6de0111a8ccab4030d034d65cf72705fc25a 100644
--- a/doc/design/images/graph_construction_example.dot
+++ b/doc/design/images/graph_construction_example.dot
@@ -44,6 +44,8 @@ digraph ImageClassificationGraph {
         OPT_W [label="SGD", color=green, shape=box];
         OPT_b [label="SGD", color=green, shape=box];
 
+        W -> OPT_W [color=green];
+        b -> OPT_b [color=green];
         d_W -> OPT_W -> W [color=green];
         d_b -> OPT_b -> b [color=green];
 
diff --git a/doc/design/images/graph_construction_example_all.png b/doc/design/images/graph_construction_example_all.png
index d769d70af759873802144f0dd0ddad21976a22eb..18d8330b60e12720bb993c8cf588d64ff8db1ea9 100644
Binary files a/doc/design/images/graph_construction_example_all.png and b/doc/design/images/graph_construction_example_all.png differ
diff --git a/doc/design/images/graph_construction_example_forward_backward.png b/doc/design/images/graph_construction_example_forward_backward.png
index 266f49d747f1278a606403f95fe8f9c815f3159e..61c3a02a04bc8891ab5b921a889829bcce386df8 100644
Binary files a/doc/design/images/graph_construction_example_forward_backward.png and b/doc/design/images/graph_construction_example_forward_backward.png differ
diff --git a/doc/design/images/graph_construction_example_forward_only.png b/doc/design/images/graph_construction_example_forward_only.png
index eb1a22da72e1d9aa7abddad1b70e1da995ebda67..14805df11fc09f64d6bc17f5e969f1400d615148 100644
Binary files a/doc/design/images/graph_construction_example_forward_only.png and b/doc/design/images/graph_construction_example_forward_only.png differ