Fixed the UnitTableViewController to properly set the datasource and delegate.