Fix argument parsing
Merge pull request #5 from moseymosey/args

Fix argument parsing