Adding code to the demo.