Minor modifications to the UI