Fixed labels for prediction model plots
Going on in the documentation of the package