Added test of y_training shape so that if it is given as a 2d array and incorrectly transposed it will be fixed, also the return values will be given in the same form as the training values