this nootbook shows our code for the toy example used for visualization in the Figure 4.

