Updated ui.js