diff --git a/data.py b/data.py
index 8620652..d5447f3 100644
--- a/data.py
+++ b/data.py
@@ -1,6 +1,7 @@
 # Data Preprocessing Template
 
 # Importing the libraries
+import tensorflow as tf
 import numpy as np
 import matplotlib.pyplot as plt
 import pandas as pd
@@ -12,5 +13,4 @@
 
 # Splitting the dataset into the Training set and Test set
 from sklearn.cross_validation import train_test_split
-X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state = 0)
-
+X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state = 0)####