fix displaying both the prediction and the original data at the same time