Change initialization method, now use UITableView as an input param