Soooo Minor UI updates