diff --git a/toolkit.py b/toolkit.py
index 9904ec7be590293503af6a515d64c599df07c115..8f10b92f0a0659cd4849ca612ebcf3936271a7f7 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -351,6 +351,7 @@ class KerasROOTClassification(object):
             sumw_bkg = np.sum(self.w_train[self.y_train == 0])
             sumw_sig = np.sum(self.w_train[self.y_train == 1])
             self._class_weight = [(sumw_sig+sumw_bkg)/(2*sumw_bkg), (sumw_sig+sumw_bkg)/(2*sumw_sig)]
+        logger.debug("Calculated class_weight: {}".format(self._class_weight))
         return self._class_weight
 
 
@@ -489,6 +490,18 @@ class KerasROOTClassification(object):
         fig.savefig(os.path.join(plot_dir, "var_{}.pdf".format(var_index)))
         plt.clf()
 
+
+    def plot_weights(self):
+        fig, ax = plt.subplots()
+        bkg = self.w_train[self.y_train == 0]
+        sig = self.w_train[self.y_train == 1]
+        ax.hist(bkg, bins=100, color="b", alpha=0.5)
+        fig.savefig(os.path.join(self.project_dir, "eventweights_bkg.pdf"))
+        fig, ax = plt.subplots()
+        ax.hist(sig, bins=100, color="r", alpha=0.5)
+        fig.savefig(os.path.join(self.project_dir, "eventweights_sig.pdf"))
+
+
     def plot_ROC(self):
 
         logger.info("Plot ROC curve")
@@ -580,7 +593,8 @@ if __name__ == "__main__":
                                 step_bkg = 100)
 
     c.load()
-    #c.train(epochs=20)
+    c.train(epochs=20)
     c.plot_ROC()
-    # c.plot_loss()
-    # c.plot_accuracy()
+    c.plot_loss()
+    c.plot_accuracy()
+    c.plot_weights()