         let authorImage: UIImage?
         let sizeDiff: Int?
         let comment: String?
-
-        init(time: String?, displayTime: String?, author: String?, authorImage: UIImage?, sizeDiff: Int?, comment: String?) {
+        var selectionThemeModel: SelectionThemeModel?
+
+        init(time: String?, displayTime: String?, author: String?, authorImage: UIImage?, sizeDiff: Int?, comment: String?, selectionThemeModel: SelectionThemeModel?) {
             self.time = time
             self.displayTime = displayTime
             self.author = author
             self.authorImage = authorImage
             self.sizeDiff = sizeDiff
             self.comment = comment
+            self.selectionThemeModel = selectionThemeModel
             super.init()
         }
     }

            case .editing:
                navigationItem.rightBarButtonItem = cancelComparisonButton
                collectionView.allowsMultipleSelection = true
                forEachVisibleCell { $0.setEditing(true) }
                compareToolbarButton.isEnabled = false
                NSLayoutConstraint.activate([
                    firstComparisonSelectionButton.widthAnchor.constraint(equalToConstant: 90),

        }
        if let button = button, let themeModel = themeModel {
            button.backgroundColor = themeModel.backgroundColor
            button.setImage(cell.authorImage, for: .normal)
            button.setTitle(cell.time, for: .normal)
            button.setTitleColor(themeModel.authorColor, for: .normal)
            button.tintColor = themeModel.authorColor
        }
        cell.selectionIndex = openSelectionIndex
        updateSelectionThemeModel(themeModel, for: cell, at: indexPath)
        cell.apply(theme: theme)

        openSelectionIndex += 1
    }
