added column check for input data
added column check for training and test data