Adding logprobs