Code to pass first test case - valid argument check