small tweaks for demo