add prediction ipynb