updated the scripts for the code demo