bug fix
Merge pull request #10 from binzh/ui

bug fix