added predictions with the test, all working good