Added check for argument