         let mediaCell = cell as! TDMediaCell
         mediaCell.didEndDisplay()
     }
-
+    
     
 }

    func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
        
        let media = mediaItems[(indexPath as NSIndexPath).item]

        let index = cart?.media.index(where: { (element) -> Bool in
            return element.asset.localIdentifier == media.asset.localIdentifier
        })

    
    
    // MARK: - Private Method(s)
    private func processReceivedMediaItems(){
        mediaItems = mediaItems.filter({ (mediaItem) -> Bool in
            if mediaItem.asset.mediaType == .image || mediaItem.asset.mediaType == .video{
                return true
            }
            return false
        })
    }
    
    private func setUpMediaCell(mediaItem: TDMediaViewModel, indexPath: IndexPath) -> TDMediaCell?{
        if mediaItem.asset.mediaType == .image{
