Remove prediction
prediction test
Add prediction parts