Update prediction.js