Allow the switch button to be customized by passing a builder that has callbacks for switching between view mode and edit mode