fix bug
Add UI