Modify demo JS