 
 import UIKit
 
-public protocol NAExpandableTableViewDataSource {
+@objc public protocol NAExpandableTableViewDataSource {
     /// Number of sections
     func numberOfExpandableSectionsInTableView(tableView: UITableView) -> Int
     

    
    private var expandDict = [Int : Bool]()
    
    public init(dataSource: NAExpandableTableViewDataSource? = nil, delegate: NAExpandableTableViewDelegate? = nil) {
        super.init()
        self.dataSource = dataSource

    
    /// Equivalent to UITableView's cellForRowAtIndexPath - called only for section title cell (the one that toggles expansion)
    func expandableTableView(tableView: UITableView, titleCellForSection section: Int, expanded: Bool) -> UITableViewCell
    
    /// The height of cells within an expandable section
    optional func expandableTableView(tableView: UITableView, heightForRowInSection section: Int) -> CGFloat
    
    /// The height of the expandable section title cell
    optional func expandableTableView(tableView: UITableView, heightForTitleCellInSection section: Int) -> CGFloat
}

@objc public protocol NAExpandableTableViewDelegate {
