modified default_prediction.ipynb