ui.js: Clean up