Add minor UI improvement.