diff --git a/TM/__init__.py b/TM/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/TM/data/__init__.py b/TM/data/__init__.py
index d77d51f..d5e29d0 100755
--- a/TM/data/__init__.py
+++ b/TM/data/__init__.py
@@ -1,4 +1,3 @@
 from . import datamgr
 from . import dataset
-from . import additional_transforms
 from . import feature_loader
diff --git a/TM/data/datamgr.py b/TM/data/datamgr.py
index f4551cd..40bd1b4 100755
--- a/TM/data/datamgr.py
+++ b/TM/data/datamgr.py
@@ -4,7 +4,6 @@
 from PIL import Image
 import numpy as np
 import torchvision.transforms as transforms
-import data.additional_transforms as add_transforms
 from data.dataset import SimpleDataset, SetDataset, EpisodicBatchSampler
 from abc import abstractmethod
 
diff --git a/TM/data/preprocess.py b/TM/data/preprocess.py
index 1b4de62..d7e63e6 100644
--- a/TM/data/preprocess.py
+++ b/TM/data/preprocess.py
@@ -5,7 +5,7 @@
 '''
 
 from anndata import read_h5ad
-import scanpy.api as sc
+import scanpy as sc
 import pandas as pd
 from collections import Counter
 import numpy as np