diff --git a/README.md b/README.md
index 5389d71..7387071 100644
--- a/README.md
+++ b/README.md
@@ -29,7 +29,11 @@ provided by PyTorch.
 
 ProtoTorch can be installed using `pip`.
 ```bash
-pip install prototorch
+pip install -U prototorch
+```
+To also install the extras, use
+```bash
+pip install -U prototorch[datasets,examples,tests]
 ```
 
 To install the bleeding-edge features and improvements:
diff --git a/examples/glvq_iris.py b/examples/glvq_iris.py
index 026ed9a..28d44ba 100644
--- a/examples/glvq_iris.py
+++ b/examples/glvq_iris.py
@@ -12,7 +12,7 @@ from prototorch.modules.prototypes import Prototypes1D
 
 # Prepare and preprocess the data
 scaler = StandardScaler()
-x_train, y_train = load_iris(True)
+x_train, y_train = load_iris(return_X_y=True)
 x_train = x_train[:, [0, 2]]
 scaler.fit(x_train)
 x_train = scaler.transform(x_train)
diff --git a/setup.py b/setup.py
index cc6370b..9e4c2f4 100644
--- a/setup.py
+++ b/setup.py
@@ -27,7 +27,7 @@ setup(name='prototorch',
           'numpy>=1.9.1',
       ],
       extras_require={
-          'datasets': ['requests'],
+          'datasets': ['requests', 'tqdm'],
           'examples': [
               'sklearn',
               'matplotlib',