Made collapse button collapse the parent element