Add some optional arguments to `plot_pred()` function to display the (multilevel) imputation method

[hanneoberman]

From plot_pred script:

TODO: add imputation method to pred plot TODO: add optional color for each possible pred matrix value TODO: add textual label for colors