before changes to predict
About to add test accuracy measures