Fix argument checking to reflect new args
Merge pull request #4 from laurence-hudson-tessella/master

Fix argument checking to reflect new args