small correction to code.js