working predict proba