some work with prediction part