Minor custom.js tweaks