Setting UICollectionDataSource as in the course