diff --git a/toolkit.py b/toolkit.py
index 2ddb7a31e246deb682fd654759e1cfe8c1c9a03f..4fce6888c9096a9577cf83a6d44339ff4aefbab5 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -419,7 +419,7 @@ class KerasROOTClassification:
         plt.xticks(np.arange(0,1,0.1))
         plt.yticks(np.arange(0,1,0.1))
         plt.legend(loc='lower left', framealpha=1.0)
-
+        plt.text(0.21,0.02,"AUC: {}".format(str(roc_auc)), size=12)
         plt.savefig(os.path.join(self.project_dir, "ROC.pdf"))
         plt.clf()
 
@@ -436,6 +436,7 @@ class KerasROOTClassification:
         plt.xlabel('epoch')
         plt.legend(['train','test'], loc='upper left')
         plt.savefig(os.path.join(self.project_dir, "losses.pdf"))
+        plt.clf()
     
 
     def plot_accuracy(self):
@@ -448,6 +449,7 @@ class KerasROOTClassification:
         plt.xlabel('epoch')
         plt.legend(['train', 'test'], loc='upper left')
         plt.savefig(os.path.join(self.project_dir, "accuracy.pdf"))
+        plt.clf()
 
 if __name__ == "__main__":