Add utility method for button state.