Few tweaks to the demo.