diff --git a/tri_nn.py b/bi_nn.py
similarity index 95%
rename from tri_nn.py
rename to bi_nn.py
index c311699..4c35901 100644
--- a/tri_nn.py
+++ b/bi_nn.py
@@ -6,7 +6,7 @@ import pandas as pd
 from os.path import exists
 
 from utils import read_csv, clean_text, get_words_from_line
-from nn import Trigrams, Model
+from nn import Bigrams, Model
 
 data = read_csv("train/in.tsv.xz")
 train_words = read_csv("train/expected.tsv")
@@ -19,7 +19,7 @@ train_data = train_data.apply(clean_text)
 vocab_size = 30000
 embed_size = 150
 
-train_dataset = Trigrams(train_data, vocab_size)
+train_dataset = Bigrams(train_data, vocab_size)
 
 ##################################################################################
 
diff --git a/nn.py b/nn.py
index 5039528..a09291f 100644
--- a/nn.py
+++ b/nn.py
@@ -4,7 +4,7 @@ from torchtext.vocab import build_vocab_from_iterator
 import itertools
 
 
-class Trigrams(torch.utils.data.IterableDataset):
+class Bigrams(torch.utils.data.IterableDataset):
     def __init__(self, data, vocabulary_size):
         self.vocab = build_vocab_from_iterator(
            get_word_lines_from_data(data),