Add cls for expanded and collapsed state