diff --git a/MANIFEST.in b/MANIFEST.in
new file mode 100644
index 0000000..0515843
--- /dev/null
+++ b/MANIFEST.in
@@ -0,0 +1 @@
+include ./*.*
diff --git a/nougat/utils/__init__.py b/nougat/utils/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/predict.py b/predict.py
index e9c021f..08a61e5 100644
--- a/predict.py
+++ b/predict.py
@@ -47,13 +47,6 @@ def get_args():
         default=None,
         help="Path to checkpoint directory",
     )
-    parser.add_argument(
-        "--pages",
-        type=int,
-        nargs="+",
-        default=None,
-        help="if only one PDF is processed, which pages? if 2 numbers are given, all in between will be processed.",
-    )
     parser.add_argument("--out", "-o", type=Path, help="Output directory.")
     parser.add_argument("--recompute", action="store_true")
     parser.add_argument("pdf", nargs="+", type=Path, help="PDF(s) to process.")
diff --git a/setup.cfg b/setup.cfg
new file mode 100644
index 0000000..0f94f37
--- /dev/null
+++ b/setup.cfg
@@ -0,0 +1,2 @@
+[metadata]
+description_file = README.md
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 7c8cbd7..d645dd4 100644
--- a/setup.py
+++ b/setup.py
@@ -40,6 +40,7 @@ def read_long_description():
             "result",
         ]
     ),
+    py_modules=["predict","app","train","test"],
     python_requires=">=3.7",
     install_requires=[
         "transformers~=4.25.1",
@@ -49,7 +50,7 @@ def read_long_description():
         "datasets[vision]",
         "pytorch-lightning>=1.8.5",
         "nltk",
-        "python-Levenshtein-wheels",
+        "python-Levenshtein",
         "sentencepiece",
         "sconf>=0.2.3",
         "albumentations",