Update prediction.java