             animateChanges {
                 self.view.layoutIfNeeded()
             }
+
+            return button
         }
     }
 

            animateChanges {
                actionButton.setButtonConfiguration(buttonConfig)
            }
        } else {
            
            removeBottonActionButton()


  // MARK: - Bottom Action Button

    @nonobjc @discardableResult
    public func addBottomActionButton(_ buttonConfig: ButtonConfiguration) -> UIButton {
    
        guard traitCollection.horizontalSizeClass == .compact else { fatalError() }
        
