Updated the source code to be able to output the prediction results.