From e139a0a0b8ff32e51e3bad99f4a6ff3dd0ef9618 Mon Sep 17 00:00:00 2001
From: Matt Bruce <matt.bruce@verizon.com>
Date: Thu, 21 Mar 2024 14:15:41 -0500
Subject: [PATCH] refactor cell some more

Signed-off-by: Matt Bruce <matt.bruce@verizon.com>
---
 .../Breadcrumbs/BreadcrumbCellItem.swift        | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/VDS/Components/Breadcrumbs/BreadcrumbCellItem.swift b/VDS/Components/Breadcrumbs/BreadcrumbCellItem.swift
index 7026c885..9f0ea68b 100644
--- a/VDS/Components/Breadcrumbs/BreadcrumbCellItem.swift
+++ b/VDS/Components/Breadcrumbs/BreadcrumbCellItem.swift
@@ -64,14 +64,31 @@ final class BreadcrumbCellItem: UICollectionViewCell {
     
     ///Updating the breadCrumbItem and UI based on the selected flag along with the surface
     func update(surface: Surface, hideSlash: Bool, breadCrumbItem: BreadcrumbItem) {
+        //remove views from stack
+        separator.removeFromSuperview()
+        self.breadCrumbItem?.removeFromSuperview()
+
+        //update surface
         separator.surface = surface
         breadCrumbItem.surface = surface
+        
+        //add to stack
         stackView.addArrangedSubview(separator)
         stackView.addArrangedSubview(breadCrumbItem)
         stackView.setCustomSpacing(VDSLayout.Spacing.space1X.value, after: separator)
+        
+        //update separator
         separator.textColor = textColorConfiguration.getColor(surface)
         separator.isHidden = hideSlash
+        
         self.breadCrumbItem = breadCrumbItem
         layoutIfNeeded()
     }
+    
+    /// Remove views from StackView.
+    override func prepareForReuse() {
+        super.prepareForReuse()
+        separator.removeFromSuperview()
+        breadCrumbItem?.removeFromSuperview()
+    }
 }