add ui
fix bug