Adding more UI elements (confirmation dialogs, etc.)