improve ui
fixed #5