UI tweak

Last commit for tonight