Add js code to demo section