Update UI minor bugs