Added CollectionViewController class for viewController