diff --git a/.gitignore b/.gitignore
index c192842d5b5201bca59cdaf155384b56bd5d5150..1c28bd7f385925cad633148103d9326b98a2d018 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,5 @@
 outputs/
+setup.sh
+run.py
 *.swp
+*.pyc
diff --git a/toolkit.py b/toolkit.py
index 1f86123c8489763d5c135c000f4c101e85367618..05dfb0e6d2fb6e01e70bad7353ab86358fd9ee81 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -353,16 +353,17 @@ class KerasROOTClassification:
         if not os.path.exists(plot_dir):
             os.mkdir(plot_dir)
         fig.savefig(os.path.join(plot_dir, "var_{}.pdf".format(var_index)))
-
+        plt.clf()
 
     def plot_ROC(self):
 
         logger.info("Plot ROC curve")
         fpr, tpr, threshold = roc_curve(self.y_test, self.scores_test, sample_weight = self.w_test)
+        fpr = 1.0 - fpr
 
         plt.grid(color='gray', linestyle='--', linewidth=1)
         plt.plot(fpr, tpr, label='NN')
-        plt.plot([0,1],[0,1], linestyle='--', color='black', label='Luck')
+        plt.plot([0,1],[1,0], linestyle='--', color='black', label='Luck')
         plt.xlabel("False positive rate (background rejection)")
         plt.ylabel("True positive rate (signal efficiency)")
         plt.title('Receiver operating characteristic')