Update src/Prediction/UI.java