TreeGrid with interactive checkbox

Is there a way to add checkbox to a TreeGrid? (vaadin 8.1)

I tried "treeGrid.setSelectionMode(SelectionMode.MULTI);" but when I select parent node, this way doesn't select automatically all its children.

Is there a way to do this?

Thank you,