Angular Material Flat Tree Parent Children Graphical Representaion

I want to show Parent Children graphical representation on Angular Material Flat Tree.

This is the design :

Here is the DEMO what i done so far.

from Angular Material Flat Tree Parent Children Graphical Representaion