Swift UIKit expanded cell not working properly

Question

I am trying to implement expandable cell inside a tableView. I am using an

expandedIndexSet : IndexSet = [] 

to keep track of the row which i