update example with better probs