Collapse not working properly when menu tree have multiple level

Problem/Motivation

Steps to reproduce

Create Menu tree with multiple level
e.g

On collapse parent item (A as on above screen shot) A1-1, A1-2, A1-3 which is under A1 is visible

Proposed resolution

On collapse parent item (A as on above screen shot) A1-1, A1-2, A1-3 which is under A1 should not display.

Remaining tasks

User interface changes

API changes

Data model changes