improve the update UI