Polish the UI: