Lunny Xiao 67b316a954 Refactor comment ()
* Refactor comment

* fix test

* improve code
..
ui.go Refactor comment ()