UI update fix

-minor bugs