improved custom.js