Fixed prediction.js