Added tests for argument checking