Updated JS for demo