diff --git a/docs/requirements.txt b/docs/requirements.txt
new file mode 100644
index 0000000..8213302
--- /dev/null
+++ b/docs/requirements.txt
@@ -0,0 +1,2 @@
+sphinx
+sphinx_rtd_theme
diff --git a/micromind/__init__.py b/micromind/__init__.py
index 2f03bb4..ad59650 100644
--- a/micromind/__init__.py
+++ b/micromind/__init__.py
@@ -2,15 +2,3 @@
 
 # Package version
 __version__ = "0.1.1"
-
-"""datasets_info is a dictionary that contains information about the attributes
-of the datasets.
-This dictionary is used in networks.py inside the from_pretrained class method
-in order to examine the inputs and initialize the PhiNet or, in case of
-mismatching between dataset and Nclasses, raise an AssertionError."""
-datasets_info = {
-    "CIFAR-100": {"Nclasses": 100, "NChannels": 3, "ext": ".pth.tar"},
-    "CIFAR-10": {"Nclasses": 10, "NChannels": 3, "ext": ".pth.tar"},
-    "ImageNet-1k": {"Nclasses": 1000, "NChannels": 3, "ext": ".pth.tar"},
-    "MNIST": {"Nclasses": 10, "NChannels": 1, "ext": ".pth.tar"},
-}