Add check for arguments