diff --git a/config.json b/config.json
index b7ade3e..982f2b0 100644
--- a/config.json
+++ b/config.json
@@ -10,9 +10,10 @@
   "max_position_embeddings": 514,
   "num_attention_heads": 16,
   "num_hidden_layers": 24,
-  "num_labels": 2,
+  "num_labels": 3,
   "output_attentions": false,
   "output_hidden_states": false,
+  "pruned_heads": {},
   "torchscript": false,
   "type_vocab_size": 1,
   "vocab_size": 50265