Refactoring to UITableViewController