add final prediction part