Adding logit option to decoding