Added support for a single command line argument; uses control flow
Merge pull request #4 from NegaMorgan/add-support-for-command-line-argument

Add support for command line argument