diff --git a/py/skewed_random_generator.py b/py/skewed_random_generator.py
index f8d99ab..5cae6e8 100644
--- a/py/skewed_random_generator.py
+++ b/py/skewed_random_generator.py
@@ -1,12 +1,6 @@
 import logging
 import random
 
-#trials = 100              # Number of total trials
-#T  = 20                    # Number of targets
-#L1 = 10                   # Number of lures (foil) similar to the (N+1)-back
-#L2 = 10                   # Number of lures (foil) similar to the (N-1)-back
-#D  = trials - L1 - L2 - T  # Number of distractors
-
 class SkewedRandomGenerator:
   """Generates random sequence of stimuli for the n-back task. Implementation is based on Ralph (2014)."""
 
@@ -14,18 +8,19 @@
 
   def __init__(self,
                N = 2, 
-               trials = 100, 
+               trials = 100,                          # Number of total trials
                alphabet=['A','B','C','D','E','F'], 
-               T = 20, 
-               L1 = 10, 
-               L2 = 10):
+               T = 20,                                # Number of targets
+               L1 = 10,                               # Number of lures (foil) similar to the (N+1)-back
+               L2 = 10                                # Number of lures (foil) similar to the (N-1)-back
+              ):
     self.N, self.alphabet, self.trials, self.T, self.L1, self.L2 = N, alphabet, trials, T, L1, L2
     self.D = trials - T - L1 - L2
 
-  def generate(self) -> str:
+  def generate(self) -> list:
     trial = 1
     self.seq = []
-    while(trial <= self.trials):
+    while trial <= self.trials:
       self.seq += self.random_stimulus(trial)
       trial += 1
     return self.seq
@@ -43,7 +38,7 @@
       self.L2 -= 1
       return self.seq[-(self.N-1)]
 
-    # distractor
+    # distract
     self.D -= 1
     alphabet = [item for item in self.alphabet if item not in self.seq[-self.N-1:-self.N+1]]
     return random.choice(alphabet)
@@ -52,4 +47,4 @@
 if __name__ == '__main__':
   generator = SkewedRandomGenerator()
   seq = generator.generate()
-  print('Skewed Random Sequence: %s' % ''.join(seq))
\ No newline at end of file
+  print('Skewed Random Sequence: %s' % ''.join(seq))