Made method to get position of highest prob.