few tweaks to the demo