Imputation and Arrays

Modified the array code and added code for imputtaion