fix Bugs

fix some bugs in UI