diff --git a/examples/run_pplm_discrim_train.py b/examples/run_pplm_discrim_train.py
index 7f10e861..9438cbba 100644
--- a/examples/run_pplm_discrim_train.py
+++ b/examples/run_pplm_discrim_train.py
@@ -72,7 +72,6 @@ class Discriminator(torch.nn.Module):
     def train_custom(self):
         for param in self.encoder.parameters():
             param.requires_grad = False
-        pass
         self.classifier_head.train()
 
     def avg_representation(self, x):
@@ -122,7 +121,7 @@ def collate_fn(data):
         padded_sequences = torch.zeros(
             len(sequences),
             max(lengths)
-        ).long()  # padding index 0
+        ).long()  # padding value = 0
 
         for i, seq in enumerate(sequences):
             end = lengths[i]