Updated code of demo

Change-Id: I2125b1e82e784b8ae71e18eae9421f768397ae62