a bit of UI polish