Add an option to hide the button