Implemented the prediction function

The COM-P prediction model might not be very correct.