Argument check now working.