diff --git a/__pycache__/__init__.cpython-36.pyc b/__pycache__/__init__.cpython-36.pyc
new file mode 100644
index 0000000..101c31b
Binary files /dev/null and b/__pycache__/__init__.cpython-36.pyc differ
diff --git a/q01_bagging/__pycache__/__init__.cpython-36.pyc b/q01_bagging/__pycache__/__init__.cpython-36.pyc
new file mode 100644
index 0000000..3d8cc6b
Binary files /dev/null and b/q01_bagging/__pycache__/__init__.cpython-36.pyc differ
diff --git a/q01_bagging/__pycache__/build.cpython-36.pyc b/q01_bagging/__pycache__/build.cpython-36.pyc
new file mode 100644
index 0000000..ddb77db
Binary files /dev/null and b/q01_bagging/__pycache__/build.cpython-36.pyc differ
diff --git a/q01_bagging/build.py b/q01_bagging/build.py
index c34fb02..8967715 100644
--- a/q01_bagging/build.py
+++ b/q01_bagging/build.py
@@ -1,3 +1,4 @@
+# %load q01_bagging/build.py
 import pandas as pd
 from sklearn.model_selection import train_test_split
 from sklearn.tree import DecisionTreeClassifier
@@ -12,7 +13,25 @@
 y = dataframe.iloc[:, -1]
 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=9)
 
+def bagging(X_train, X_test, y_train, y_test, n_est):
+    acc_score_train = []
+    acc_score_test = []
+    for i in range(1,n_est+1):
+        bagging_clf = BaggingClassifier(DecisionTreeClassifier(), n_estimators=i, max_samples=0.67,max_features = 0.67,
+                                bootstrap=True, random_state=9)
+        bagging_clf.fit(X_train, y_train)
 
-# Write your code here
+        y_test_pred = bagging_clf.predict(X_test)
+        y_train_pred = bagging_clf.predict(X_train)
+
+        acc_score_train.append(accuracy_score(y_train, y_train_pred))
+        acc_score_test.append(accuracy_score(y_test, y_test_pred))
+        
+    plt.plot(range(1,n_est+1),acc_score_train, label = 'Train Set')
+    plt.plot(range(1,n_est+1),acc_score_test, label = 'Test Set')
+    plt.legend()
+    plt.xlabel('No. of estimators')
+    plt.ylabel('Accuracy')
+    plt.show()
 
 
diff --git a/q01_bagging/tests/__pycache__/__init__.cpython-36.pyc b/q01_bagging/tests/__pycache__/__init__.cpython-36.pyc
new file mode 100644
index 0000000..68ee5f2
Binary files /dev/null and b/q01_bagging/tests/__pycache__/__init__.cpython-36.pyc differ
diff --git a/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc b/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc
new file mode 100644
index 0000000..d5a89be
Binary files /dev/null and b/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc differ