Changes to predictOneVsAll