The demo works for the linear mural network. I need to polish the geometry of the nn