Clean up script.js