update UI

update ui and fix code