Update ui
Minor reformatting