Minor UI alignment fixes