Make small demo tweaks