Cosmetic update

UI changes.