make sure sum and mean along column
 fix notation of z_and_x
pass decision tree