Update custom.js for 3.0