def add_oov_padding(w2idx, oov="<UNK>", pad="<PAD>"):
    """Add OOV and PAD tokens to w2idx dict."""
    w2idx = {w: idx + 2 for w, idx in w2idx.items()}
    w2idx.update({pad: 0, oov: 1})
    return w2idx