Add selection logic for CollectionVC