Added the code for this demo

Added the code for this demo