Custom Matrix Visual - Expand/collapse row headers problems

Hello, I've been having some trouble figuring out how to do expand and collapse on row headers of my custom matrix visual. I've followed the guide on your docs here but buttons don't all work.   I...