Concise way of checking arguments with python list comprehension feature
some modification
Merge pull request #10 from shahinism/master

Some small modification!