Some UI polish.