     
     
     // 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{

        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
        })
