Wee bit of UI

UI tweaks