Add UI improvement.