diff --git a/app.py b/app.py
index dadd4e6..fffd376 100644
--- a/app.py
+++ b/app.py
@@ -8,7 +8,7 @@ from sklearn.metrics import classification_report
   
 # Function importing Dataset 
 def importdata(): 
-    dataset = pd.read_csv( 'Train.csv',sep= ',', header = None) 
+    dataset = pd.read_csv('Train.csv',sep= ',', header = None) 
     return dataset 
   
 # Function to split the dataset 
@@ -16,7 +16,7 @@ def splitdataset(dataset):
   
     # Separating the target variable 
     X = dataset.values[:, 1:5] 
-    Y = dataset.values[:, 0] 
+    Y = dataset.values[:,-1] 
   
     # Splitting the dataset into train and test 
     X_train, X_test, y_train, y_test = train_test_split(  
@@ -74,7 +74,8 @@ def main():
       
     # Building Phase 
     data = importdata() 
-    X, Y, X_train, X_test, y_train, y_test = splitdataset(data) 
+    X, Y, X_train, X_test, y_train, y_test = splitdataset(data)
+    
     clf_gini = train_using_gini(X_train, X_test, y_train) 
     clf_entropy = tarin_using_entropy(X_train, X_test, y_train)