prediction for regression is now working (more or less)