add info to plot

Add prob of model in post proc plot