changed to correct prediction format (array)