Update flattenImage.js