Updated UI.js for 77a46cf
Fix errors for #4