added the extraction of prediciton data