diff --git a/README.md b/README.md
index 6b914a4aab3c2ca8068e5ec7bc0dbac38f0961c4..1099b912c5a5e072d631f4cf4338b298e09a8163 100644
--- a/README.md
+++ b/README.md
@@ -29,8 +29,7 @@ Output:
 To install package, run
 
 ```
-pip install -r requirements.txt
-pip install -r requirements_dev.txt
+pip install .
 ```
 
 
diff --git a/setup.py b/setup.py
index abcfdb7c1e60131eb9689d4de0f0c8a7eef80574..3fe12debd8988acce22bf44e9f942c2051348894 100644
--- a/setup.py
+++ b/setup.py
@@ -1,22 +1,28 @@
 """Set up project."""
-from setuptools import setup, find_packages
 from pathlib import Path
+from setuptools import setup, find_packages
 
 project_root_dir = Path(__file__).parent.resolve()
 with open(project_root_dir / "requirements.txt",
           "r", encoding="utf-8") as f:
     INSTALL_REQUIRES = f.read().splitlines()
 
-url = 'https://git.scicore.unibas.ch/zavolan_group/tools/terminal-fragment-selector'
+URL = ('https://git.scicore.unibas.ch/zavolan_group/'
+       'tools/terminal-fragment-selector')
 
 setup(
     name='terminal-fragment-selector',
     version='0.1.1',
-    url=url,
+    url=URL,
     license='MIT',
     author='Hugo Madge Leon, Sunho Kim, Tanya Nandan',
     author_email='hmadge@ethz.ch',
     description='Terminal fragment selector',
     packages=find_packages(),
-    install_requires=INSTALL_REQUIRES
+    install_requires=INSTALL_REQUIRES,
+    entry_points={
+        'console_scripts': [
+            'terminal-fragment-selector=term_frag_sel.cli:main'
+            ]
+        }
 )
diff --git a/term_frag_sel/cli.py b/term_frag_sel/cli.py
index 67b9f7d554b0f2376955a2c77338d2b726cce1e4..37188c9077b0cdfd4b5f016cb32d94ee49bc571d 100644
--- a/term_frag_sel/cli.py
+++ b/term_frag_sel/cli.py
@@ -18,13 +18,15 @@ logging.basicConfig(
 logger = logging.getLogger("main")
 
 
-def main(args: argparse.Namespace):
+def main():
     """Use CLI arguments to fragment sequences and output text file \
     with selected terminal fragments.
 
     Args:
         args (parser): list of arguments from CLI.
     """
+    args = parse_arguments()
+
     if not isinstance(args, argparse.Namespace):
         raise TypeError("Input should be argparse.Namespace")
 
@@ -38,8 +40,10 @@ def main(args: argparse.Namespace):
     logger.info("Fragmentation of %s...", args.fasta)
     splits = np.arange(0, len(list(fasta))+args.size, args.size)
 
-    for i, split in enumerate(splits):
-        fasta_dict = fasta[split:splits[i+1]]
+    for i in range(len(splits) - 1):
+        split = splits[i]
+        keys = list(fasta.keys())[split:splits[i+1]]
+        fasta_dict = {key: fasta[key] for key in keys}
         term_frags = fragmentation(fasta_dict, seq_counts,
                                    args.mean, args.std)
 
@@ -132,6 +136,4 @@ if __name__ == '__main__':
         level=logging.INFO,
     )
     logger = logging.getLogger(__name__)
-
-    arguments = parse_arguments()
-    main(arguments)
+    main()
diff --git a/term_frag_sel/utils.py b/term_frag_sel/utils.py
index a565dcefe272521ce141f5925f047cf7c761d51b..083513945034ff8d7694bd0332496b075198f745 100644
--- a/term_frag_sel/utils.py
+++ b/term_frag_sel/utils.py
@@ -23,5 +23,4 @@ def check_positive(value: str) -> int:
     except ValueError as exc:
         raise argparse.ArgumentTypeError(f"""Expected positive integer,
                                          got: {value}""") from exc
-    else:
-        return ivalue
+    return ivalue
diff --git a/tests/test_cli.py b/tests/test_cli.py
index 4ffe56aa4c4a9dfc43f195c305dc4ada3e89536d..ca946126c84e1328e0a42de24151efb49fb742f8 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -27,5 +27,5 @@ def test_file():
 
 def test_main():
     """Test main() function."""
-    with pytest.raises(TypeError):
-        main("")
+    with pytest.raises(SystemExit):
+        main()