Fix issue with model_splits() function

[Sum02dean]

The method needs to be adapted to take into account the labels column now being part of the design matrix.