Fix bugs

Change ui

Signed-off-by: Jianmin Zhou <toandrew@infthink.com>