ios-swift-collapsible-table-section | simple iOS Swift project demonstrates how to implement | iOS library

I am using CollapsibleTableView from here and modified it as per my requirement to achieve collapsible sections. Here is how it looks now.

Since there is a border for my section as per the UI design, I had chosen the section header to be my UI element that holds data in both collapsed and expanded modes.

Reason: I tried but couldn't get it working in this model explained below -

** Have my header elements in section header and details of each item in its cell. By default, the section is in collapsed state. When user taps on the header, the cell is toggled to display. As I said, since there is a border that needs to be shown to the whole section (tapped header and its cell), I chose section header to be my UI element of operation. Here is my code for tableView -

I'm going through this tutorial (https://github.com/jeantimex/ios-swift-collapsible-table-section) trying to build an expandable listview into my application. The only differance between my code and the tutorials is I'm doing it with a storyboard.

I seem to have it almost fully functional but I have one problem. The rows are collapsing and then are repopulating without being clicked a second time. I've gone through my code multiple times but I can't seem to find the problem.

Here is my code:

List View Controller Class:

Following the answer of jeantimex on how to expand/collapse a section here and his github, I added the following codes to hide the rows when a section is tapped: