latest update, adding in prediction stuff here