Using the ui prop for button