diff --git a/python/src/ai/network.py b/python/src/ai/network.py
index 928ecac..ae0345a 100644
--- a/python/src/ai/network.py
+++ b/python/src/ai/network.py
@@ -1,5 +1,8 @@
 import json
 import os
+import time
+import timeit
+
 import tensorflow as tf
 import sklearn
 import numpy as np
@@ -83,6 +86,7 @@ class AI:
         self.__dict = None
         self.__network = None
         self.__sources = None
+        self._phrases = None
         self.filename = ''
 
     def add_source(self, text):
@@ -116,7 +120,7 @@ class AI:
                                    np.zeros([len(word_list) - len(phrase)]))
                 phrases.append(phrase)
         phrases = np.array(phrases)
-        print(phrases.shape)
+        self._phrases = phrases
 
     def active_train(self):
         pass