Some updates to the notebook. Added separate prediction script.