Add possibility to set the button tag