Cleaned up custom.js