diff --git a/section_4.1_compute_visualize.py b/section_4.1_compute_visualize.py
index 3f7e0e7f8931f177791dbed8722fbb66adf7887e..56acdfedbdb03452078ca70458cc746380248a06 100644
--- a/section_4.1_compute_visualize.py
+++ b/section_4.1_compute_visualize.py
@@ -12,6 +12,7 @@ import numpy as np
 from matplotlib import pyplot as plt
 from scipy import io
 import os
+import matplotlib
 
 
 np.random.seed(1)
@@ -70,6 +71,8 @@ io.savemat(path_for_results + name_dataset + '.mat',
 
 
 ## Some plots
+matplotlib.use('TKAgg')
+
 R = np.linspace(0, 1, int(1.5*array_cv_f.shape[1]))
 array_color=plt.cm.hsv(R)
 
@@ -103,7 +106,7 @@ plt.legend(list_legend + ['A[0,1]', 'A[1,0]'])
 plt.title('matrix entries')
 plt.xscale('log')
 plt.xlabel('iterations')
-plt.draw()
+plt.show()
 
 
 
diff --git a/setup.sh b/setup.sh
index a454aebb685974f8f2504604865a66ab1fd36a87..ddd3cd6cfbae20d02328d5c8268b4f1cdf7de2b9 100755
--- a/setup.sh
+++ b/setup.sh
@@ -16,8 +16,8 @@ source venv/bin/activate
 pip install --upgrade pip
 pip3 install --ignore-installed torch==1.9.1+cpu torchvision==0.10.1+cpu torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
 
-scipy==1.7.3
-sklearn==0.0.post5
+pip3 install scipy==1.7.3
+pip3 install scikit-learn==1.0.2
 
 
 # Install further stuff