Added option to add/remove a button to the right corner of the header