more tweaks for the demo