Improve demo js