UPDATE from master for PlotPrediction.py plot_prediction.py