Edit custom.js with Andrew's code