Add argument-count check