Update to demo

Minor fix for demo buttons