Move JS into setupOutput() in script.js