Added changes to inverse_pred_script.py to implement plot changes