Collapse/expand working for components tree but not classes