add collapse buttons working again.