bit of UI upgrade