Updated argument logic.
Merge pull request #32 from VMBindraban/hotfix/button-logic

Updated argument logic.