Hi, There is a `class BertForMultipleChoice` in `modeling_bert.py` in transformers library. Does it do the same as the model in notebook? Thanks!