Minor UI space fixing