BERT in document classification