Minor change in UI