diff --git a/32_manuscript.pdf b/32_manuscript.pdf
new file mode 100644
index 0000000..5b0aa9e
Binary files /dev/null and b/32_manuscript.pdf differ
diff --git a/prepare_data.py b/prepare_data.py
index 4af5cf3..737c91e 100644
--- a/prepare_data.py
+++ b/prepare_data.py
@@ -31,7 +31,7 @@ def prepare_folds(args):
     for cat in categories:
         for doc in cat:
             folds[f].append(doc)
-            f = (f + 1) % 5
+            f = (f + 1) % docs_per_fold
 
     # list[num_folds, docs_per_fold] --> list[num_folds * docs_per_fold]
     idx_order = sum(folds, [])