func collectionView(_ collectionView: UICollectionView,
  didSelectItemAt indexPath: IndexPath) {
    //code..
  }