Small changes to script.js