Changes to plotting and prediction model files