diff --git a/requirements.txt b/requirements.txt
index 7e090cff537e7ad33e48a89da8c2685e965510b2..daffb1ca4dd6d51ca3b5f9c882e2790708e5b374 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,10 +1,18 @@
-gradio==3.31.0
-h5py==3.8.0
-localthickness==0.1.2
-matplotlib==3.7.1
-numpy==1.24.3
-outputformat==0.1.3
-plotly==5.14.1
-scipy==1.10.1
-setuptools==59.6.0
-tifffile==2023.4.12
+albumentations>=1.3.1
+gradio>=3.44.3
+h5py>=3.9.0
+localthickness>=0.1.2
+matplotlib>=3.7.1
+monai>=1.2.0
+numpy>=1.25.2
+outputformat>=0.1.3
+Pillow>=10.0.1
+plotly>=5.14.1
+scipy>=1.11.2
+seaborn>=0.12.2
+setuptools>=68.0.0
+tifffile>=2023.4.12
+torch>=2.0.1
+torchvision>=0.15.2
+torchinfo>=1.8.0
+tqdm>=4.65.0
diff --git a/setup.py b/setup.py
index bc58133bb8c05a6ede388413415d37dedbe9010e..f0521d7456f40eb6a64e252b1775be1e8fc870b7 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@ with open("README.md", "r", encoding="utf-8") as f:
 
 setup(
     name="qim3d",
-    version="0.1.3",
+    version="0.2.0",
     author="Felipe Delestro",
     author_email="fima@dtu.dk",
     description="QIM tools and user interfaces",
@@ -21,7 +21,7 @@ setup(
     classifiers=[
         "License :: OSI Approved :: MIT License",
         "Programming Language :: Python :: 3",
-        "Development Status :: 1 - Planning",
+        "Development Status :: 2 - Pre-Alpha",
         "Intended Audience :: Education",
         "Intended Audience :: Science/Research",
         "Natural Language :: English",
@@ -31,14 +31,24 @@ setup(
         "Topic :: Software Development :: User Interfaces",
     ],
     python_requires=">=3.6",
-    install_requires=['gradio>=3.31.0',
-                      'h5py>=3.8.0',
-                      'localthickness>=0.1.2',
-                      'matplotlib>=3.7.1',
-                      'numpy>=1.24.3',
-                      'outputformat>=0.1.3',
-                      'plotly>=5.14.1',
-                      'scipy>=1.10.1',
-                      'setuptools>=59.6.0',
-                      'tifffile>=2023.4.12']
+    install_requires=[
+        "albumentations>=1.3.1",
+        "gradio>=3.44.3",
+        "h5py>=3.9.0",
+        "localthickness>=0.1.2",
+        "matplotlib>=3.7.1",
+        "monai>=1.2.0",
+        "numpy>=1.25.2",
+        "outputformat>=0.1.3",
+        "Pillow>=10.0.1",
+        "plotly>=5.14.1",
+        "scipy>=1.11.2",
+        "seaborn>=0.12.2",
+        "setuptools>=68.0.0",
+        "tifffile>=2023.4.12",
+        "torch>=2.0.1",
+        "torchvision>=0.15.2",
+        "torchinfo>=1.8.0",
+        "tqdm>=4.65.0",
+    ],
 )