Fixed ui bug
Merge pull request #21 from windyarts/master

UI bug fix