Small fixes to get OctanePrediction.ipynb to work