Extended the button demo