add the icons and modify the UI code