update prediction part