add highest prob func