main input arg

[jbcuglietta]

issue:
passing a number not divisible by the size of the array in data_input will cause the main to halt.

fix: input arg checking against the size of the input matrix