Check argument size

Check if the size of the input is smaller than the flag given; set the flag to the input size if true.