Change API of segmented-button component