adding the code which will give the prediction of each source