Make the button more configurable