A little change

As it is
Merge pull request #7 from zfaustk/UI

A little change