Move JS to custom.js