further developed the prediction generator