Add argument check