Minor updates in UI