UI improvement
Merge pull request #2 from vmotsak/master

UI improvement