Update screenshot.js