update UI

more polish