Minor UI adjustment