Minor changes in JS