Do better on validating input argumnet