diff --git a/scAR-gpu.yml b/scAR-gpu.yml
index 79bc182..aa00162 100644
--- a/scAR-gpu.yml
+++ b/scAR-gpu.yml
@@ -7,7 +7,7 @@ dependencies:
 - conda-forge::python>=3.8.6
 - nvidia::cudatoolkit>=11.1
 - pytorch::pytorch>=1.10.0
-- pytorch::tcorchvision>=0.9.0
+- pytorch::torchvision>=0.9.0
 - pytorch::torchaudio>=0.8.0
 - pytorch-mutex=*=cuda
 - conda-forge::tqdm>=4.62.3
diff --git a/scAR/__init__.py b/scAR/__init__.py
index 94a49f0..2cb1272 100644
--- a/scAR/__init__.py
+++ b/scAR/__init__.py
@@ -1 +1,4 @@
-from scAR.main._scAR import model
\ No newline at end of file
+# -*- coding: utf-8 -*-
+
+from scAR.main._scAR import model
+from scAR.main import _data_generater as DataSimulator
\ No newline at end of file
diff --git a/scAR/main/__init__.py b/scAR/main/__init__.py
index a04d931..aedc925 100644
--- a/scAR/main/__init__.py
+++ b/scAR/main/__init__.py
@@ -1,5 +1,2 @@
 # -*- coding: utf-8 -*-
-
-__version__ = '0.2.0'
-
-from scAR.main import _data_generater as DataSimulator
\ No newline at end of file
+__version__ = '0.2.1'
\ No newline at end of file
diff --git a/setup.py b/setup.py
index b4a6d0a..5574fd4 100644
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,7 @@
     entry_points = {
         'console_scripts': ['scar=scAR.main.__main__:main'],},
     include_package_data=True,
-    url='https://github.com/CaibinSh/scAR',
+    url='https://github.com/Novartis/scAR',
     license='MIT',
     install_requires=[
         "torch>=1.10.0",