Added prediction process for test data.