Add icon, UI tweaks