Controlling UI, Simplified

What is short, is simple.
DOH! Missed a semi-colon!
Merge pull request #8 from jailuthra/patch-1

Code simplified and shortened (minor change) - Thanks Jai :)