add highest prob func
Merge pull request #9 from brunops/get_highest_prob

add highest prob func