add prediction and change training and test