Small modifications in UI.