Add argument checks