Update js/ui.js