Update script.js - fix bug