diff --git a/exercises/4-Numpy/numpy1-chessboard-solution.py b/exercises/4-Numpy/numpy1-chessboard-solution.py
index 651c838b4015ec9ff477b27df78c5e8152f83744..671df41879846138390174c0fe5b32d43d3f930f 100644
--- a/exercises/4-Numpy/numpy1-chessboard-solution.py
+++ b/exercises/4-Numpy/numpy1-chessboard-solution.py
@@ -3,7 +3,8 @@
# Exercise 4.1: Numpy chessboard - Solution
#
-# Create a 8x8 matrix with a chessboard pattern with numbers 1 and 0.
+# Create a 8x8 matrix with a chessboard pattern with numbers 1 and 0,
+# then visualize it with the function imshow from matplotlib.
import numpy as np
import matplotlib.pyplot as plt
@@ -19,5 +20,4 @@ def chessboard():
M = chessboard()
-print(M)
plt.imshow(M, cmap='binary')
diff --git a/exercises/4-Numpy/numpy1-chessboard.py b/exercises/4-Numpy/numpy1-chessboard.py
index 9c114d41ec74b7f1feb7551ee08a6dd19e29f4e2..c165490d4cc85d7e6b05dc030293d428e2865bd6 100644
--- a/exercises/4-Numpy/numpy1-chessboard.py
+++ b/exercises/4-Numpy/numpy1-chessboard.py
@@ -3,12 +3,16 @@
# Exercise 4.1: Numpy chessboard
#
-# Create a 8x8 matrix with a chessboard pattern with numbers 1 and 0.
+# Create a 8x8 matrix with a chessboard pattern with numbers 1 and 0,
+# then visualize it with the function imshow from matplotlib.
import numpy as np
+import matplotlib.pyplot as plt
+
def chessboard():
pass # Your code here
-print(chessboard())
+M = chessboard()
+plt.imshow(M, cmap='binary')