Adds support for UITableViewController and UICollectionViewController