updated checkArgument and checkState statements