Added code for demo test in demo
Added code for demo test in demo