We should check that X is 2D, or at least make it more explicit

[aloctavodia]

We currently mention that X is "a covariate matrix", and Y a "response vector", but we could be more explicit about the expected shapes.