improve ui
Merge pull request #4 from HarrisonChen/ui

improve ui