Changed: UITableViewDataSource to view model