DEV: Make the segmented control button to work